test: test superintend n_samples

ebonnal · Aug 7, 2023 · fa16ca8 · fa16ca8
1 parent fb89617
commit fa16ca8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/kioss/pipe.py b/kioss/pipe.py
@@ -320,9 +320,7 @@ def iterate():
 
         return timeit.timeit(iterate, number=1)
 
-    def superintend(
-        self, n_samples: int = 0, n_error_samples: int = 8
-    ) -> List[T]:
+    def superintend(self, n_samples: int = 0, n_error_samples: int = 8) -> List[T]:
         """
         Superintend the Pipe: iterate over the pipe until it is exhausted and raise a RuntimeError if any exceptions occur during iteration.
 

diff --git a/test/test_pipe.py b/test/test_pipe.py
@@ -421,7 +421,7 @@ def test_superintend(self):
             ValueError,
             lambda: Pipe("12-3").map(int).superintend(),
         )
-        self.assertListEqual(Pipe("123").map(int).superintend(n_samples=3), [1, 2, 3])
+        self.assertListEqual(Pipe("123").map(int).superintend(n_samples=2), [1, 2])
 
     def test_log(self):
         self.assertListEqual(