Changeset d48cc19 in sasview for src/sas/qtgui/Perspectives/Fitting

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

Location:

src/sas/qtgui/Perspectives/Fitting

Files:

• FittingUtilities.py (modified) (1 diff)
• FittingWidget.py (modified) (10 diffs)
• UnitTesting/FittingWidgetTest.py (modified) (7 diffs)

src/sas/qtgui/Perspectives/Fitting/FittingUtilities.py

@@ -262 +262 @@
         residuals.y = -y
     else:
+        # TODO: fix case where applying new data from file on top of existing model data
         y = (fn - gn[index][0]) / en
         residuals.y = y

src/sas/qtgui/Perspectives/Fitting/FittingWidget.py

@@ -74 +74 @@
         self.weighting = 0
 
+        # Data for chosen model
+        self.model_data = None
+
         # Which tab is this widget displayed in?
         self.tab_id = id
@@ -250 +253 @@
         """
         self.cmdFit.setEnabled(False)
-        self.cmdPlot.setEnabled(True)
+        self.cmdPlot.setEnabled(False)
         self.options_widget.cmdComputePoints.setVisible(False) # probably redundant
         self.chkPolydispersity.setEnabled(True)
@@ -305 +308 @@
         self.has_error_column = False
 
+        # Set enablement on calculate/plot
+        self.cmdPlot.setEnabled(True)
+
         # SasModel -> QModel
         self.SASModelToQModel(model)
 
         if self.data_is_loaded:
+            self.cmdPlot.setText("Show Plot")
             self.calculateQGridForModel()
         else:
+            self.cmdPlot.setText("Calculate")
             # Create default datasets if no data passed
             self.createDefaultDataset()
@@ -569 +577 @@
         Plot the current set of data
         """
+        # Regardless of previous state, this should now be `plot show` functionality only
+        self.cmdPlot.setText("Show Plot")
+        self.recalculatePlotData()
+        self.showPlot()
+
+    def recalculatePlotData(self):
+        """
+        Generate a new dataset for model
+        """
         if not self.data_is_loaded:
             self.createDefaultDataset()
         self.calculateQGridForModel()
+
+    def showPlot(self):
+        """
+        Show the current plot in MPL
+        """
+        # Show the chart if ready
+        data_to_show = self.data if self.data_is_loaded else self.model_data
+        if data_to_show is not None:
+            self.communicate.plotRequestedSignal.emit([data_to_show])
 
     def onOptionsUpdate(self):
@@ -582 +608 @@
         self.lblMinRangeDef.setText(str(self.q_range_min))
         self.lblMaxRangeDef.setText(str(self.q_range_max))
-        self.onPlot()
+        self.recalculatePlotData()
 
     def setDefaultStructureCombo(self):
@@ -766 +792 @@
         # Force the chart update when actual parameters changed
         if model_column == 1:
-            self.onPlot()
+            self.recalculatePlotData()
 
     def checkboxSelected(self, item):
@@ -891 +917 @@
         Plot the current 1D data
         """
-        fitted_plot = self.logic.new1DPlot(return_data, self.tab_id)
-        self.calculateResiduals(fitted_plot)
+        fitted_data = self.logic.new1DPlot(return_data, self.tab_id)
+        self.calculateResiduals(fitted_data)
+        self.model_data = fitted_data
 
     def complete2D(self, return_data):
@@ -900 +927 @@
         fitted_data = self.logic.new2DPlot(return_data)
         self.calculateResiduals(fitted_data)
+        self.model_data = fitted_data
 
     def calculateResiduals(self, fitted_data):
@@ -914 +942 @@
         self.lblChi2Value.setText(chi2_repr)
 
+        self.communicate.plotUpdateSignal.emit([fitted_data])
+
         # Plot residuals if actual data
         if self.data_is_loaded:
@@ -920 +950 @@
             self.createNewIndex(residuals_plot)
             self.communicate.plotUpdateSignal.emit([residuals_plot])
-
-        self.communicate.plotUpdateSignal.emit([fitted_data])
 
     def calcException(self, etype, value, tb):

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__":