feat: more commands (#109)

iterpy/iter.py

@@ -51,7 +51,7 @@ def __getitem__(self, index: int | slice) -> T | "Iter[T]":
             )
 
     def __repr__(self) -> str:
-        return f"Iter({self._nonconsumable_iterable})"
+        return f"Iter{self._nonconsumable_iterable}"
 
     def __eq__(self, other: object) -> bool:
         if not isinstance(other, Iter):
@@ -154,3 +154,15 @@ def unique_by(self, func: Callable[[T], S]) -> "Iter[T]":
                 values.append(value)
 
         return Iter(values)
+
+    def enumerate(self) -> "Iter[tuple[int, T]]":
+        return Iter(enumerate(self._iterator))
+
+    def find(self, func: Callable[[T], bool]) -> T | None:
+        for value in self._iterator:
+            if func(value):
+                return value
+        return None
+
+    def last(self) -> T:
+        return self._nonconsumable_iterable[-1]

iterpy/iter.pyi

@@ -42,6 +42,10 @@ class Iter(Generic[T]):
     def all(self, func: Callable[[T], bool]) -> bool: ...
     def unique(self) -> Iter[T]: ...
     def unique_by(self, func: Callable[[T], U]) -> Iter[T]: ...
+    def enumerate(self) -> Iter[tuple[int, T]]: ...
+    def find(self, func: Callable[[T], bool]) -> T: ...
+    def last(self) -> T: ...
+    def sum(self) -> T: ...
     def groupby(
         self, func: Callable[[T], str]
     ) -> Iter[tuple[str, list[T]]]: ...

iterpy/test_iter.py

@@ -123,6 +123,26 @@ def test_unique_by():
     ]
 
 
+def test_enumerate():
+    test_iterator = Iter([1, 2, 3])
+    assert test_iterator.enumerate().to_list() == [
+        (0, 1),
+        (1, 2),
+        (2, 3),
+    ]
+
+
+def test_find():
+    test_iterator = Iter([1, 2, 3])
+    assert test_iterator.find(lambda x: x == 2) == 2
+    assert test_iterator.find(lambda x: x == 4) is None
+
+
+def test_last():
+    test_iterator = Iter([1, 2, 3])
+    assert test_iterator.last() == 3
+
+
 def test_flatten():
     test_input: list[list[int]] = [[1, 2], [3, 4]]
     iterator = Iter(test_input)