Changeset d48cc19 in sasview for src/sas/qtgui/Perspectives/Fitting/UnitTesting/FittingWidgetTest.py

Timestamp:

May 2, 2017 9:04:48 AM (8 years ago)

Author:

Piotr Rozyczko <rozyczko@…>

Branches:

ESS_GUI, ESS_GUI_Docs, ESS_GUI_batch_fitting, ESS_GUI_bumps_abstraction, ESS_GUI_iss1116, ESS_GUI_iss879, ESS_GUI_iss959, ESS_GUI_opencl, ESS_GUI_ordering, ESS_GUI_sync_sascalc

Children:

02ddfb4

Parents:

0a6e097

Message:

Compute/Show? Plot button logic: SASVIEW-271
Unit tests for plotting in fitting: SASVIEW-501

File:

• src/sas/qtgui/Perspectives/Fitting/UnitTesting/FittingWidgetTest.py (modified) (7 diffs)

src/sas/qtgui/Perspectives/Fitting/UnitTesting/FittingWidgetTest.py

@@ -1 +1 @@
 import sys
 import unittest
+import time
 
 from PyQt4 import QtGui
@@ -21 +22 @@
 
 class dummy_manager(object):
-    def communicate(self):
-        return Communicate()
+    communicate = Communicate()
 
 class FittingWidgetTest(unittest.TestCase):
@@ -80 +80 @@
         self.assertFalse(widget_with_data.acceptsData())
 
-    def notestSelectModel(self):
+    def testSelectModel(self):
         """
         Test if models have been loaded properly
-        :return:
         """
         fittingWindow =  self.widget
@@ -280 +279 @@
         pass
 
-    def notestCreateTheoryIndex(self):
+    def testCreateTheoryIndex(self):
         """
         Test the data->QIndex conversion
@@ -292 +291 @@
 
         # Create the index
-        self.widget.createTheoryIndex()
-
-        # Check the signal sent
-        print spy
-
-        # Check the index
+        self.widget.createTheoryIndex(Data1D(x=[1,2], y=[1,2]))
+
+        # Make sure the signal has been emitted
+        self.assertEqual(spy.count(), 1)
+
+        # Check the argument type
+        self.assertIsInstance(spy.called()[0]['args'][0], QtGui.QStandardItem)
 
     def testCalculateQGridForModel(self):
@@ -305 +305 @@
         # Mock the thread creation
         threads.deferToThread = MagicMock()
+        # Model for theory
+        self.widget.SASModelToQModel("cylinder")
         # Call the tested method
         self.widget.calculateQGridForModel()
+        time.sleep(1)
         # Test the mock
         self.assertTrue(threads.deferToThread.called)
         self.assertEqual(threads.deferToThread.call_args_list[0][0][0].__name__, "compute")
 
-    def testComplete1D(self):
-        """
-        Check that a new 1D plot is generated
-        """
-        # TODO when chisqr method coded
-        pass
-
-    def testComplete2D(self):
-        """
-        Check that a new 2D plot is generated
-        """
-        # TODO when chisqr method coded
-        pass
+    def testCalculateResiduals(self):
+        """
+        Check that the residuals are calculated and plots updated
+        """
+        test_data = Data1D(x=[1,2], y=[1,2])
+
+        # Model for theory
+        self.widget.SASModelToQModel("cylinder")
+        # Invoke the tested method
+        self.widget.calculateResiduals(test_data)
+        # Check the Chi2 value - should be undetermined
+        self.assertEqual(self.widget.lblChi2Value.text(), '---')
+
+        # Force same data into logic
+        self.widget.logic.data = test_data
+        self.widget.calculateResiduals(test_data)
+        # Now, the difference is 0, as data is the same
+        self.assertEqual(self.widget.lblChi2Value.text(), '0')
+
+        # Change data
+        test_data_2 = Data1D(x=[1,2], y=[2.1,3.49])
+        self.widget.logic.data = test_data_2
+        self.widget.calculateResiduals(test_data)
+        # Now, the difference is non-zero
+        self.assertEqual(float(self.widget.lblChi2Value.text()), 1.715)
 
     def testSetPolyModel(self):
@@ -410 +425 @@
         self.assertEqual(self.widget._model_model.rowCount(), last_row)
 
+    def testPlotTheory(self):
+        """
+        See that theory item can produce a chart
+        """
+        # By default, the compute/plot button is disabled
+        self.assertFalse(self.widget.cmdPlot.isEnabled())
+        self.assertEqual(self.widget.cmdPlot.text(), 'Show Plot')
+
+        # Assign a model
+        self.widget.show()
+        # Change the category index so we have a model available
+        category_index = self.widget.cbCategory.findText("Sphere")
+        self.widget.cbCategory.setCurrentIndex(category_index)
+
+        # Check the enablement/text
+        self.assertTrue(self.widget.cmdPlot.isEnabled())
+        self.assertEqual(self.widget.cmdPlot.text(), 'Calculate')
+
+        # Spying on plot update signal
+        spy = QtSignalSpy(self.widget, self.widget.communicate.plotRequestedSignal)
+
+        # Press Calculate
+        QtTest.QTest.mouseClick(self.widget.cmdPlot, QtCore.Qt.LeftButton)
+
+        # Observe cmdPlot caption change
+        self.assertEqual(self.widget.cmdPlot.text(), 'Show Plot')
+
+        # Make sure the signal has NOT been emitted
+        self.assertEqual(spy.count(), 0)
+
+        # Click again
+        QtTest.QTest.mouseClick(self.widget.cmdPlot, QtCore.Qt.LeftButton)
+
+        # This time, we got the update signal
+        self.assertEqual(spy.count(), 0)
+
+    def testPlotData(self):
+        """
+        See that data item can produce a chart
+        """
+        # By default, the compute/plot button is disabled
+        self.assertFalse(self.widget.cmdPlot.isEnabled())
+        self.assertEqual(self.widget.cmdPlot.text(), 'Show Plot')
+
+        self.widget.show()
+
+        # Set data
+        test_data = Data1D(x=[1,2], y=[1,2])
+
+        # Force same data into logic
+        self.widget.logic.data = test_data
+        self.widget.data_is_loaded = True
+
+        # Change the category index so we have a model available
+        category_index = self.widget.cbCategory.findText("Sphere")
+        self.widget.cbCategory.setCurrentIndex(category_index)
+
+        # Check the enablement/text
+        self.assertTrue(self.widget.cmdPlot.isEnabled())
+        self.assertEqual(self.widget.cmdPlot.text(), 'Show Plot')
+
+        # Spying on plot update signal
+        spy = QtSignalSpy(self.widget, self.widget.communicate.plotRequestedSignal)
+
+        # Press Calculate
+        QtTest.QTest.mouseClick(self.widget.cmdPlot, QtCore.Qt.LeftButton)
+
+        # Observe cmdPlot caption did not change
+        self.assertEqual(self.widget.cmdPlot.text(), 'Show Plot')
+
+        # Make sure the signal has been emitted == new plot
+        self.assertEqual(spy.count(), 1)
+
 
 if __name__ == "__main__":