update waterfall test for bragg peak update #REF766

mantidproject · Jun 8, 2022 · a2f94d9 · a2f94d9
1 parent 25d49dc
commit a2f94d9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mslice/tests/cut_plot_test.py b/mslice/tests/cut_plot_test.py
@@ -100,6 +100,7 @@ def test_update_legend(self):
 
     def test_waterfall(self):
         self.cut_plot._apply_offset = MagicMock()
+        self.cut_plot.update_bragg_peaks = MagicMock()
         self.cut_plot.waterfall = True
         self.cut_plot.waterfall_x = 1
         self.cut_plot.waterfall_y = 2
@@ -108,3 +109,4 @@ def test_waterfall(self):
         self.cut_plot.waterfall = False
         self.cut_plot.toggle_waterfall()
         self.cut_plot._apply_offset.assert_called_with(0, 0)
+        self.cut_plot.update_bragg_peaks.assert_called_with(refresh=True)