Changeset 1042dba in sasview for src/sas/qtgui/UnitTesting

Timestamp:

Jun 17, 2016 9:56:01 AM (8 years ago)

Author:

Piotr Rozyczko <piotr.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:

9e426c1

Parents:

a281ab8

Message:

Plottables support in the data object. Additional menu items.

Location:

src/sas/qtgui/UnitTesting

Files:

• DataExplorerTest.py (modified) (2 diffs)
• GuiUtilsTest.py (modified) (3 diffs)

src/sas/qtgui/UnitTesting/DataExplorerTest.py

@@ -9 +9 @@
 # Local
 from sas.sasgui.guiframe.dataFitting import Data1D
+from sas.sascalc.dataloader.loader import Loader
+from sas.sasgui.guiframe.data_manager import DataManager
+
 from DataExplorer import DataExplorerWindow
 from GuiManager import GuiManager
 from GuiUtils import *
 from UnitTesting.TestUtils import QtSignalSpy
+from Plotter import Plotter
 
 app = QApplication(sys.argv)
@@ -297 +301 @@
         self.assertTrue(self.form.manager.add_data.called)
 
+    def testNewPlot(self):
+        """
+        Creating new plots from Data1D/2D
+        """
+        loader = Loader()
+        manager = DataManager()
+
+        # get Data1D
+        p_file="cyl_400_20.txt"
+        output_object = loader.load(p_file)
+        new_data = [manager.create_gui_data(output_object, p_file)]
+
+        # Mask the plot show call
+        Plotter.show = MagicMock()
+
+        # Mask retrieval of the data
+        self.form.plotsFromCheckedItems = MagicMock(return_value=new_data)
+
+        # Call the plotting method
+        self.form.newPlot()
+
+        # The plot was displayed
+        self.assertTrue(Plotter.show.called)
+
 
 if __name__ == "__main__":

src/sas/qtgui/UnitTesting/GuiUtilsTest.py

@@ -1 +1 @@
 import sys
 import unittest
+
+from PyQt4 import QtCore
+from PyQt4 import QtGui
+
+# SV imports
+from sas.sascalc.dataloader.loader import Loader
+from sas.sasgui.guiframe.data_manager import DataManager
 
 # Tested module
@@ -12 +19 @@
 
     def tearDown(self):
-        '''empty'''
+        '''Empty'''
         pass
 
@@ -76 +83 @@
         self.assertEqual(str(list_from_item[1]), test_list[1])
 
+
+    def testPlotsFromCheckedItems(self):
+        """
+        Test addition of a plottable to the model
+        """
+
+        # Mockup data
+        test_list0 = "FRIDAY"
+        test_list1 = "SATURDAY"
+        test_list2 = "MONDAY"
+
+        # Main item ("file")
+        checkbox_model = QtGui.QStandardItemModel()
+        checkbox_item = QtGui.QStandardItem(True)
+        checkbox_item.setCheckable(True)
+        checkbox_item.setCheckState(QtCore.Qt.Checked)
+        test_item0 = QtGui.QStandardItem()
+        test_item0.setData(QtCore.QVariant(test_list0))
+
+        # Checked item 1
+        test_item1 = QtGui.QStandardItem(True)
+        test_item1.setCheckable(True)
+        test_item1.setCheckState(QtCore.Qt.Checked)
+        object_item = QtGui.QStandardItem()
+        object_item.setData(QtCore.QVariant(test_list1))
+        test_item1.setChild(0, object_item)
+
+        checkbox_item.setChild(0, test_item0)
+        checkbox_item.appendRow(test_item1)
+
+        # Unchecked item 2
+        test_item2 = QtGui.QStandardItem(True)
+        test_item2.setCheckable(True)
+        test_item2.setCheckState(QtCore.Qt.Unchecked)
+        object_item = QtGui.QStandardItem()
+        object_item.setData(QtCore.QVariant(test_list2))
+        test_item2.setChild(0, object_item)
+        checkbox_item.appendRow(test_item2)
+
+        checkbox_model.appendRow(checkbox_item)
+
+        # Pull out the "plottable" documents
+        plot_list = plotsFromCheckedItems(checkbox_model)
+
+        # Make sure only the checked data is present
+        # FRIDAY IN
+        self.assertIn(test_list0, plot_list)
+        # SATURDAY IN
+        self.assertIn(test_list1, plot_list)
+        # MONDAY NOT IN
+        self.assertNotIn(test_list2, plot_list)
+
+    def testInfoFromData(self):
+        """
+        Test Info element extraction from a plottable object
+        """
+        loader = Loader()
+        manager = DataManager()
+
+        # get Data1D
+        p_file="cyl_400_20.txt"
+        output_object = loader.load(p_file)
+        new_data = manager.create_gui_data(output_object, p_file)
+
+        # Extract Info elements into a model item
+        item = infoFromData(new_data)
+
+        # Test the item and its children
+        self.assertIsInstance(item, QtGui.QStandardItem)
+        self.assertEqual(item.rowCount(), 5)
+        self.assertEqual(item.text(), "Info")
+        self.assertIn(p_file,   item.child(0).text())
+        self.assertIn("Run",    item.child(1).text())
+        self.assertIn("Data1D", item.child(2).text())
+        self.assertIn(p_file,   item.child(3).text())
+        self.assertIn("Process",item.child(4).text())
+
+
 if __name__ == "__main__":
     unittest.main()