Fix test

xsuite · Jan 30, 2025 · 570b001 · 570b001
1 parent ab3164b
commit 570b001
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_line.py b/tests/test_line.py
@@ -81,9 +81,10 @@ def test_repeated_elements_survey():
     # https://github.com/xsuite/xplt/issues/31
     line = xt.Line(elements={"obm": xt.Bend(length=0.5)}, element_names=["obm", "obm"])
     # line.replace_all_repeated_elements()
-    print(line.survey())
+    survey = line.survey()
+    print(survey)
 
-    plot = xplt.FloorPlot(line=line)
+    plot = xplt.FloorPlot(survey)
 
     boxes = plot.artists_boxes
     assert_equal(len(boxes), 2)