Changes in src/sas/qtgui/Plotting/UnitTesting/PlotterTest.py [5b144c6:b8080e1] in sasview

File:

• src/sas/qtgui/Plotting/UnitTesting/PlotterTest.py (modified) (11 diffs)

src/sas/qtgui/Plotting/UnitTesting/PlotterTest.py

@@ -57 +57 @@
         self.plotter.data = self.data
         self.plotter.show()
-        FigureCanvas.draw = MagicMock()
+        FigureCanvas.draw_idle = MagicMock()
 
         self.plotter.plot(hide_error=False)
 
         self.assertEqual(self.plotter.ax.get_xscale(), 'log')
-        self.assertTrue(FigureCanvas.draw.called)
+        self.assertTrue(FigureCanvas.draw_idle.called)
 
         self.plotter.figure.clf()
@@ -70 +70 @@
         self.plotter.data = self.data
         self.plotter.show()
-        FigureCanvas.draw = MagicMock()
+        FigureCanvas.draw_idle = MagicMock()
 
         self.plotter.plot(hide_error=True)
 
         self.assertEqual(self.plotter.ax.get_yscale(), 'log')
-        self.assertTrue(FigureCanvas.draw.called)
+        self.assertTrue(FigureCanvas.draw_idle.called)
         self.plotter.figure.clf()
 
@@ -91 +91 @@
         self.plotter.data = data
         self.plotter.show()
-        FigureCanvas.draw = MagicMock()
+        FigureCanvas.draw_idle = MagicMock()
 
         self.plotter.plot(hide_error=True)
@@ -97 +97 @@
         self.assertEqual(self.plotter.ax.get_xscale(), 'linear')
         self.assertEqual(self.plotter.ax.get_yscale(), 'linear')
-        self.assertTrue(FigureCanvas.draw.called)
+        self.assertTrue(FigureCanvas.draw_idle.called)
 
     def testCreateContextMenuQuick(self):
@@ -262 +262 @@
         # Just this one plot
         self.assertEqual(len(list(self.plotter.plot_dict.keys())), 1)
-        self.plotter.onLinearFit('Test name')
+        self.plotter.onLinearFit(1)
 
         # Check that exec_ got called
@@ -289 +289 @@
 
         # Delete one set
-        self.plotter.onRemovePlot('Test name 2')
+        self.plotter.onRemovePlot(2)
         # Assure we have two sets
         self.assertEqual(len(list(self.plotter.plot_dict.keys())), 1)
@@ -296 +296 @@
 
         # Delete the remaining set
-        self.plotter.onRemovePlot('Test name')
+        self.plotter.onRemovePlot(1)
         # Assure we have no plots
         self.assertEqual(len(list(self.plotter.plot_dict.keys())), 0)
@@ -331 +331 @@
         self.assertEqual(yl, "$YAXIS(cake)$")
         # The hide_error flag should also remain
-        self.assertTrue(self.plotter.plot_dict['Test name 2'].hide_error)
+        self.assertTrue(self.plotter.plot_dict[2].hide_error)
         self.plotter.figure.clf()
 
@@ -355 +355 @@
 
         # Reverse the toggle
-        self.plotter.onToggleHideError('Test name 2')
+        self.plotter.onToggleHideError(2)
         # See that the labels didn't change
         xl = self.plotter.ax.xaxis.label.get_text()
@@ -362 +362 @@
         self.assertEqual(yl, "$YAXIS(cake)$")
         # The hide_error flag should toggle
-        self.assertEqual(self.plotter.plot_dict['Test name 2'].hide_error, not error_status)
+        self.assertEqual(self.plotter.plot_dict[2].hide_error, not error_status)
         self.plotter.figure.clf()
 
@@ -417 +417 @@
         self.assertEqual(yl, "$YAXIS(cake)$")
         # The hide_error flag should be as set
-        self.assertEqual(self.plotter.plot_dict['Test name 2'].hide_error, error_status)
+        self.assertEqual(self.plotter.plot_dict[2].hide_error, error_status)
         self.plotter.figure.clf()