Support any and all

zhukovgreen · Nov 29, 2024 · 045ab59 · 045ab59
1 parent 5462e79
commit 045ab59
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/friendly_sequences/__init__.py b/friendly_sequences/__init__.py
@@ -181,6 +181,16 @@ def join(
     ) -> str:
         return with_.join(self)
 
+    def all(
+        self: Seq[T],
+    ) -> bool:
+        return all(self)
+
+    def any(
+        self: Seq[T],
+    ) -> bool:
+        return any(self)
+
     def __iter__(  # noqa: PYI034
         self: Seq[T],
     ) -> Iterator[T]:

diff --git a/tests/test_friendly_sequences.py b/tests/test_friendly_sequences.py
@@ -31,6 +31,10 @@ def add_one(i: int) -> int:
         .reduce(lambda left, right: left + right)
     ) == "1234"
 
+    assert Seq((1, 2, 3)).map(lambda x: x // 2 == 0).all() is False
+    assert Seq((2, 4, 6)).map(lambda x: x // 2 == 0).all() is False
+    assert Seq((1, 2, 3)).map(lambda x: x // 2 == 0).any() is True
+
 
 def test_chaining():
     def filter_expr(i: int) -> TypeGuard[int]: