fix failing tests

lcviz/plugins/binning/binning.py

@@ -140,7 +140,8 @@ def _toggle_marks(self, event={}):
 
             mark.visible = this_visible
 
-        if visible:
+        if visible and event.get('name', '') in ('is_active', 'show_live_preview'):
+            # then the marks themselves need to be updated
             self._live_update(event)
 
     @observe('dataset_selected', 'ephemeris_selected',
@@ -152,6 +153,10 @@ def _live_update(self, event={}):
             self.bin_enabled = self.n_bins != '' and self.n_bins > 0
             return
 
+        if event.get('name', '') not in ('is_active', 'show_live_preview'):
+            # mark visibility hasn't been handled yet
+            self._toggle_marks()
+
         try:
             lc = self.bin(add_data=False)
         except Exception:

lcviz/plugins/flatten/flatten.py

@@ -170,7 +170,8 @@ def _toggle_marks(self, event={}):
         for mark in flattened_marks.values():
             mark.visible = visible
 
-        if visible:
+        if visible and event.get('name') in ('is_active', 'show_live_preview'):
+            # then the marks themselves need to be updated
             self._live_update(event)
 
     @observe('dataset_selected',
@@ -186,6 +187,10 @@ def _live_update(self, event={}):
             return
         self.flatten_err = ''
 
+        if event.get('name') not in ('is_active', 'show_live_preview'):
+            # mark visibility hasn't been handled yet
+            self._toggle_marks(event)
+
         if self.unnormalize:
             output_flux = output_lc.flux.value
         else:

lcviz/tests/test_plugin_flatten.py

@@ -54,7 +54,9 @@ def test_plugin_flatten(helper, light_curve_like_kepler_quarter):
         # update polyorder (live-preview should re-appear and have changed from before)
         f.polyorder = before_polyorder + 1
         assert f._obj.flatten_err == ''
-        after_update = _get_marks_from_viewer(tv)[0].y
+        marks = _get_marks_from_viewer(tv)
+        assert len(marks) == 2
+        after_update = marks[0].y
         assert not np.allclose(before_update, after_update)
 
         orig_label = f.dataset.selected