Fix reproducibility tests being skipped due to iterating over filter …

…object twice
xsuite · Jan 21, 2025 · cde8527 · cde8527
1 parent a25a5be
commit cde8527
Show file tree

Hide file tree

Showing 3 changed files with 53 additions and 10 deletions.
diff --git a/examples/line.ipynb b/examples/line.ipynb
diff --git a/examples/twiss.ipynb b/examples/twiss.ipynb
diff --git a/tests/test_examples.py b/tests/test_examples.py
@@ -63,7 +63,8 @@ def keep(out):
                 return False
             return True
 
-        reference_outputs = filter(keep, reference_cell.get("outputs", []))
+        # Make sure these are lists, since filter object can be iterated over only once!
+        reference_outputs = list(filter(keep, reference_cell.get("outputs", [])))
         actual_outputs = list(filter(keep, actual_cell.get("outputs", [])))
 
         print("reference_outputs:", [o.get("output_type") for o in reference_outputs])