Fix test of Exercise 2, Input-Output (empa-scientific-it#91)

baffelli · edoardob90 · web-flow · commit baf18d383654 · 2023-04-25T10:11:47.000Z
Co-authored-by: Simone Baffelli &lt;simone.baffelli@empa.ch&gt;
Co-authored-by: Edoardo Baldi &lt;edoardob90@gmail.com&gt;
diff --git a/tutorial/tests/test_input_output.py b/tutorial/tests/test_input_output.py
@@ -27,10 +27,10 @@ def test_exercise1(function_to_test):
 
 def reference_solution_exercise2(f: pl.Path)-> int:
     with open(f) as lines:
-        return len(lines.readlines())
+        return len(list(itertools.chain.from_iterable([l.split() for l in lines.readlines()])))
 
 def test_exercise2(function_to_test):
-    f = get_data("example.csv")
+    f = get_data("lines.txt")
     assert function_to_test(f) == reference_solution_exercise2(f)
 
 
