make stitch test more robust (#114)

by ensuring the data is loaded in a viewer before checking if the stitch plugin is active
spacetelescope · Apr 22, 2024 · ccfb8ce · ccfb8ce
1 parent 8196447
commit ccfb8ce
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lcviz/tests/test_plugin_stitch.py b/lcviz/tests/test_plugin_stitch.py
@@ -16,11 +16,12 @@ def test_docs_snippets(helper, light_curve_like_kepler_quarter):
 
 
 def test_plugin_stitch(helper, light_curve_like_kepler_quarter):
-    helper.load_data(light_curve_like_kepler_quarter)
+    helper.load_data(light_curve_like_kepler_quarter, 'lc1')
 
     assert "Stitch" not in helper.plugins.keys()
 
-    helper.load_data(light_curve_like_kepler_quarter.copy())
+    helper.load_data(light_curve_like_kepler_quarter.copy(), 'lc2')
+    helper.app.add_data_to_viewer('flux-vs-time', 'lc2')
     assert "Stitch" in helper.plugins.keys()
 
     stitch = helper.plugins['Stitch']