add __call__ as alias for .count

streamable/stream.py

@@ -139,6 +139,16 @@ def count(self) -> int:
 
         return sum(1 for _ in self)
 
+    def __call__(self) -> int:
+        """
+        Iterates over this stream until exhaustion and returns the count of elements.
+        Equivalent to `stream.count()`.
+
+        Returns:
+            int: Number of elements yielded during an entire iteration over this stream.
+        """
+        return self.count()
+
     def display(self, level: int = logging.INFO) -> "Stream[T]":
         """
         Logs (INFO level) a representation of the stream.

tests/test_stream.py

@@ -1356,10 +1356,15 @@ def effect(x: int) -> None:
             l.append(x)
 
         stream = Stream(lambda: map(effect, src))
-        self.assertIs(
+        self.assertEqual(
             stream.count(),
             N,
-            msg="`exhaust` should return self.",
+            msg="`count` should return the count of elements.",
+        )
+        self.assertEqual(
+            stream(),
+            N,
+            msg="`__call__` should return the count of elements.",
         )
         self.assertListEqual(
             l, list(src), msg="`exhaust` should iterate over the entire stream."