Add startswith() and endswith() methods to Path

earthsnake/path.py

@@ -75,3 +75,13 @@ class Path:
 
     def __repr__(self) -> str:
         return f'<Path {self.path}>'
+
+    def startswith(self, sub: str) -> bool:
+        """Check if path starts with sub"""
+
+        return self.path.startswith(sub)
+
+    def endswith(self, sub: str) -> bool:
+        """Check if path ends with sub"""
+
+        return self.path.endswith(sub)

tests/test_path.py

@@ -106,3 +106,21 @@ def test_repr() -> None:
     """Test the __repr__ method"""
 
     assert repr(Path('/test.txt')) == '<Path /test.txt>'
+
+
+def test_startswith() -> None:
+    """Test the startswith method"""
+
+    path = Path('/test.txt')
+
+    assert path.startswith('/test')
+    assert not path.startswith('test')
+
+
+def test_endswith() -> None:
+    """Test the endswith method"""
+
+    path = Path('/test.txt')
+
+    assert path.endswith('.txt')
+    assert not path.endswith('.py')