🗓 Jan 6, 2024 1:44:03 PM

✨ drop_bytes method
🧪 tests added/updated

.gitignore

@@ -1,3 +1,9 @@
+bin/
+pyvenv.cfg
+.envrc
+hexfile.bin
+share/
+
 .bandit.json
 .scannerwork
 .DS_Store

chepy/modules/utils.py

@@ -693,7 +693,7 @@ def shuffle(self) -> UtilsT:
         unchanged if any other types.
 
         Returns:
-            Chepy: _description_
+            Chepy: The Chepy object
         """
         data = self.state
         if not isinstance(
@@ -716,3 +716,27 @@ def shuffle(self) -> UtilsT:
         random.shuffle(data)
         self.state = data
         return self
+
+    @ChepyDecorators.call_stack
+    def drop_bytes(self, start: int, length: int) -> UtilsT:
+        """Drop bytes from starting index up to length
+
+        Args:
+            start (int): Starting index
+            length (int): Number of bytes to drop
+
+        Raises:
+            ValueError: If start or length < -1
+
+        Returns:
+            Chepy: The Chepy object
+        """
+        if start < 0 or length < 0:
+            raise ValueError(
+                "Start and length must be non-negative integers."
+            )  # pragma: no cover
+
+        end = start + length
+        data = self._convert_to_bytes()
+        self.state = data[:start] + data[end:]
+        return self

chepy/modules/utils.pyi

@@ -38,3 +38,4 @@ class Utils(ChepyCore):
     def set(self: UtilsT) -> UtilsT: ...
     def regex_to_str(self: UtilsT, all_combo: bool=...) -> UtilsT: ...
     def shuffle(self: UtilsT) -> UtilsT: ...
+    def drop_bytes(self: UtilsT, start: int, length: int) -> UtilsT: ...

tests/test_utils.py

@@ -236,3 +236,7 @@ def test_shuffle():
     assert len(Chepy([1, 2, 3]).shuffle().o) == 3
     assert len(Chepy("abc").shuffle().o) == 3
     assert len(Chepy(b"abdc").shuffle().o) != 3
+
+
+def test_drop_bytes():
+    assert Chepy("hello").drop_bytes(2, 2).o == b"heo"