test_count: fix

ebonnal · Oct 16, 2024 · 05595d6 · 05595d6
1 parent 7727125
commit 05595d6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_stream.py b/tests/test_stream.py
@@ -1361,14 +1361,14 @@ def effect(x: int) -> None:
             N,
             msg="`count` should return the count of elements.",
         )
+        self.assertListEqual(
+            l, list(src), msg="`count` should iterate over the entire stream."
+        )
         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."
-        )
 
     def test_multiple_iterations(self) -> None:
         stream = Stream(src)