source: sasview/src/sas/qtgui/MainWindow/UnitTesting/DataExplorerTest.py @ d6b8a1d

ESS_GUIESS_GUI_DocsESS_GUI_batch_fittingESS_GUI_bumps_abstractionESS_GUI_iss1116ESS_GUI_iss879ESS_GUI_iss959ESS_GUI_openclESS_GUI_orderingESS_GUI_sync_sascalc
Last change on this file since d6b8a1d was 7fb471d, checked in by Piotr Rozyczko <rozyczko@…>, 6 years ago

Update for unit tests and minor functionality quirks
  • Property mode set to 100755
File size: 27.3 KB
Line 
1import sys
2import unittest
3
4from PyQt4.QtGui import *
5from PyQt4.QtTest import QTest
6from PyQt4.QtCore import *
7from unittest.mock import MagicMock
8from unittest.mock import patch
9from mpl_toolkits.mplot3d import Axes3D
10
11# set up import paths
12import path_prepare
13
14# Local
15from sas.qtgui.Plotting.PlotterData import Data1D
16from sas.sascalc.dataloader.loader import Loader
17from sas.qtgui.MainWindow.DataManager import DataManager
18
19from sas.qtgui.MainWindow.DataExplorer import DataExplorerWindow
20from sas.qtgui.MainWindow.GuiManager import GuiManager
21from sas.qtgui.Utilities.GuiUtils import *
22from UnitTesting.TestUtils import QtSignalSpy
23from sas.qtgui.Plotting.Plotter import Plotter
24from sas.qtgui.Plotting.Plotter2D import Plotter2D
25import sas.qtgui.Plotting.PlotHelper as PlotHelper
26
27if not QApplication.instance():
28    app = QApplication(sys.argv)
29
30class DataExplorerTest(unittest.TestCase):
31    '''Test the Data Explorer GUI'''
32    def setUp(self):
33        '''Create the GUI'''
34        class MyPerspective(object):
35            def communicator(self):
36                return Communicate()
37            def allowBatch(self):
38                return False
39            def setData(self, data_item=None, is_batch=False):
40                return None
41            def title(self):
42                return "Dummy Perspective"
43
44        class dummy_manager(object):
45            def communicator(self):
46                return Communicate()
47            def perspective(self):
48                return MyPerspective()
49            def workspace(self):
50                return None
51
52        self.form = DataExplorerWindow(None, dummy_manager())
53
54    def tearDown(self):
55        '''Destroy the GUI'''
56        self.form.close()
57        self.form = None
58
59    def testDefaults(self):
60        '''Test the GUI in its default state'''
61        # Tab widget
62        self.assertIsInstance(self.form, QTabWidget)
63        self.assertEqual(self.form.count(), 2)
64
65        # Buttons - data tab
66        self.assertEqual(self.form.cmdLoad.text(), "Load data")
67        self.assertEqual(self.form.cmdDeleteData.text(), "Delete")
68        self.assertEqual(self.form.cmdDeleteTheory.text(), "Delete")
69        self.assertEqual(self.form.cmdFreeze.text(), "Freeze Theory")
70        self.assertEqual(self.form.cmdSendTo.text(), "Send data to")
71        self.assertEqual(self.form.cmdSendTo.iconSize(), QSize(48, 48))
72        self.assertIsInstance(self.form.cmdSendTo.icon(), QIcon)
73        self.assertEqual(self.form.chkBatch.text(), "Batch mode")
74        self.assertFalse(self.form.chkBatch.isChecked())
75
76        # Buttons - theory tab
77
78        # Combo boxes
79        self.assertEqual(self.form.cbSelect.count(), 6)
80        self.assertEqual(self.form.cbSelect.currentIndex(), 0)
81
82        # Models - data
83        self.assertIsInstance(self.form.model, QStandardItemModel)
84        self.assertEqual(self.form.treeView.model().rowCount(), 0)
85        self.assertEqual(self.form.treeView.model().columnCount(), 0)
86        self.assertEqual(self.form.model.rowCount(), 0)
87        self.assertEqual(self.form.model.columnCount(), 0)
88        self.assertIsInstance(self.form.data_proxy, QSortFilterProxyModel)
89        self.assertEqual(self.form.data_proxy.sourceModel(), self.form.model)
90        self.assertEqual("[^()]", str(self.form.data_proxy.filterRegExp().pattern()))
91        self.assertIsInstance(self.form.treeView, QTreeView)
92
93        # Models - theory
94        self.assertIsInstance(self.form.theory_model, QStandardItemModel)
95        self.assertEqual(self.form.freezeView.model().rowCount(), 0)
96        self.assertEqual(self.form.freezeView.model().columnCount(), 0)
97        self.assertEqual(self.form.theory_model.rowCount(), 0)
98        self.assertEqual(self.form.theory_model.columnCount(), 0)
99        self.assertIsInstance(self.form.theory_proxy, QSortFilterProxyModel)
100        self.assertEqual(self.form.theory_proxy.sourceModel(), self.form.theory_model)
101        self.assertEqual("[^()]", str(self.form.theory_proxy.filterRegExp().pattern()))
102        self.assertIsInstance(self.form.freezeView, QTreeView)
103
104    def testWidgets(self):
105        """
106        Test if all required widgets got added
107        """
108    def testLoadButton(self):
109        loadButton = self.form.cmdLoad
110
111        filename = "cyl_400_20.txt"
112        # Initialize signal spy instances
113        spy_file_read = QtSignalSpy(self.form, self.form.communicator.fileReadSignal)
114
115        # Return no files.
116        QtGui.QFileDialog.getOpenFileNames = MagicMock(return_value=None)
117
118        # Click on the Load button
119        QTest.mouseClick(loadButton, Qt.LeftButton)
120
121        # Test the getOpenFileName() dialog called once
122        self.assertTrue(QtGui.QFileDialog.getOpenFileNames.called)
123        QtGui.QFileDialog.getOpenFileNames.assert_called_once()
124
125        # Make sure the signal has not been emitted
126        self.assertEqual(spy_file_read.count(), 0)
127
128        # Now, return a single file
129        QtGui.QFileDialog.getOpenFileNames = MagicMock(return_value=filename)
130
131        # Click on the Load button
132        QTest.mouseClick(loadButton, Qt.LeftButton)
133        QtGui.qApp.processEvents()
134
135        # Test the getOpenFileName() dialog called once
136        self.assertTrue(QtGui.QFileDialog.getOpenFileNames.called)
137        QtGui.QFileDialog.getOpenFileNames.assert_called_once()
138
139        # Expected one spy instance
140        #self.assertEqual(spy_file_read.count(), 1)
141        #self.assertIn(filename, str(spy_file_read.called()[0]['args'][0]))
142
143    def testLoadFiles(self):
144        """
145        Test progress bar update while loading of multiple files
146        """
147        # Set up the spy on progress bar update signal
148        spy_progress_bar_update = QtSignalSpy(self.form,
149            self.form.communicator.progressBarUpdateSignal)
150
151        # Populate the model
152        filename = ["cyl_400_20.txt", "P123_D2O_10_percent.dat", "cyl_400_20.txt"]
153        self.form.readData(filename)
154
155        # 0, 0, 33, 66, -1 -> 5 signals reaching progressBar
156        self.assertEqual(spy_progress_bar_update.count(), 5)
157
158        expected_list = [0, 0, 33, 66, -1]
159        spied_list = [spy_progress_bar_update.called()[i]['args'][0] for i in range(5)]
160        self.assertEqual(expected_list, spied_list)
161       
162    def testDeleteButton(self):
163        """
164        Functionality of the delete button
165        """
166        deleteButton = self.form.cmdDeleteData
167
168        # Mock the confirmation dialog with return=No
169        QtGui.QMessageBox.question = MagicMock(return_value=QtGui.QMessageBox.No)
170
171        # Populate the model
172        filename = ["cyl_400_20.txt", "cyl_400_20.txt", "cyl_400_20.txt"]
173        self.form.readData(filename)
174
175        # Assure the model contains three items
176        self.assertEqual(self.form.model.rowCount(), 3)
177
178        # Assure the checkboxes are on
179        item1 = self.form.model.item(0)
180        item2 = self.form.model.item(1)
181        item3 = self.form.model.item(2)
182        self.assertTrue(item1.checkState() == QtCore.Qt.Checked)
183        self.assertTrue(item2.checkState() == QtCore.Qt.Checked)
184        self.assertTrue(item3.checkState() == QtCore.Qt.Checked)
185
186        # Click on the delete  button
187        QTest.mouseClick(deleteButton, Qt.LeftButton)
188
189        # Test the warning dialog called once
190        self.assertTrue(QtGui.QMessageBox.question.called)
191
192        # Assure the model still contains the items
193        self.assertEqual(self.form.model.rowCount(), 3)
194
195        # Now, mock the confirmation dialog with return=Yes
196        QtGui.QMessageBox.question = MagicMock(return_value=QtGui.QMessageBox.Yes)
197
198        # Click on the delete  button
199        QTest.mouseClick(deleteButton, Qt.LeftButton)
200
201        # Test the warning dialog called once
202        self.assertTrue(QtGui.QMessageBox.question.called)
203
204        # Assure the model contains no items
205        self.assertEqual(self.form.model.rowCount(), 0)
206
207        # Click delete once again to assure no nasty behaviour on empty model
208        QTest.mouseClick(deleteButton, Qt.LeftButton)
209
210    def testDeleteTheory(self):
211        """
212        Test that clicking "Delete" in theories tab removes selected indices
213        """
214        deleteButton = self.form.cmdDeleteTheory
215
216        # Mock the confirmation dialog with return=No
217        QtGui.QMessageBox.question = MagicMock(return_value=QtGui.QMessageBox.No)
218
219        # Populate the model
220        item1 = QtGui.QStandardItem(True)
221        item1.setCheckable(True)
222        item1.setCheckState(QtCore.Qt.Checked)
223        item1.setText("item 1")
224        self.form.theory_model.appendRow(item1)
225        item2 = QtGui.QStandardItem(True)
226        item2.setCheckable(True)
227        item2.setCheckState(QtCore.Qt.Unchecked)
228        item2.setText("item 2")
229        self.form.theory_model.appendRow(item2)
230
231        # Assure the model contains two items
232        self.assertEqual(self.form.theory_model.rowCount(), 2)
233
234        # Assure the checkboxes are on
235        self.assertTrue(item1.checkState() == QtCore.Qt.Checked)
236        self.assertTrue(item2.checkState() == QtCore.Qt.Unchecked)
237
238        # Click on the delete  button
239        QTest.mouseClick(deleteButton, Qt.LeftButton)
240
241        # Test the warning dialog called once
242        self.assertTrue(QtGui.QMessageBox.question.called)
243
244        # Assure the model still contains the items
245        self.assertEqual(self.form.theory_model.rowCount(), 2)
246
247        # Now, mock the confirmation dialog with return=Yes
248        QtGui.QMessageBox.question = MagicMock(return_value=QtGui.QMessageBox.Yes)
249
250        # Click on the delete  button
251        QTest.mouseClick(deleteButton, Qt.LeftButton)
252
253        # Test the warning dialog called once
254        self.assertTrue(QtGui.QMessageBox.question.called)
255
256        # Assure the model contains 1 item
257        self.assertEqual(self.form.theory_model.rowCount(), 1)
258
259        # Set the remaining item to checked
260        self.form.theory_model.item(0).setCheckState(QtCore.Qt.Checked)
261
262        # Click on the delete button again
263        QTest.mouseClick(deleteButton, Qt.LeftButton)
264
265        # Assure the model contains no items
266        self.assertEqual(self.form.theory_model.rowCount(), 0)
267
268        # Click delete once again to assure no nasty behaviour on empty model
269        QTest.mouseClick(deleteButton, Qt.LeftButton)
270
271
272    def testSendToButton(self):
273        """
274        Test that clicking the Send To button sends checked data to a perspective
275        """
276        # Send empty data
277        mocked_perspective = self.form.parent.perspective()
278        mocked_perspective.setData = MagicMock()
279
280        # Click on the Send To  button
281        QTest.mouseClick(self.form.cmdSendTo, Qt.LeftButton)
282
283        # The set_data method not called
284        self.assertFalse(mocked_perspective.setData.called)
285               
286        # Populate the model
287        filename = ["cyl_400_20.txt"]
288        self.form.readData(filename)
289
290        # setData is the method we want to see called
291        mocked_perspective = self.form.parent.perspective()
292        mocked_perspective.setData = MagicMock(filename)
293
294        # Assure the checkbox is on
295        self.form.cbSelect.setCurrentIndex(0)
296
297        # Click on the Send To  button
298        QTest.mouseClick(self.form.cmdSendTo, Qt.LeftButton)
299
300        # Test the set_data method called once
301        #self.assertTrue(mocked_perspective.setData.called)
302
303        # open another file
304        filename = ["cyl_400_20.txt"]
305        self.form.readData(filename)
306
307        # Mock the warning message
308        QtGui.QMessageBox = MagicMock()
309
310        # Click on the button
311        QTest.mouseClick(self.form.cmdSendTo, Qt.LeftButton)
312
313        # Assure the message box popped up
314        QtGui.QMessageBox.assert_called_once()
315
316    def testDataSelection(self):
317        """
318        Tests the functionality of the Selection Option combobox
319        """
320        # Populate the model with 1d and 2d data
321        filename = ["cyl_400_20.txt", "P123_D2O_10_percent.dat"]
322        self.form.readData(filename)
323
324        # Unselect all data
325        self.form.cbSelect.setCurrentIndex(1)
326
327        self.form.show()
328        app.exec_()
329
330        # Test the current selection
331        item1D = self.form.model.item(0)
332        item2D = self.form.model.item(1)
333        self.assertTrue(item1D.checkState() == QtCore.Qt.Unchecked)
334        self.assertTrue(item2D.checkState() == QtCore.Qt.Unchecked)       
335
336        # Select all data
337        self.form.cbSelect.setCurrentIndex(0)
338
339        # Test the current selection
340        self.assertTrue(item1D.checkState() == QtCore.Qt.Checked)
341        self.assertTrue(item2D.checkState() == QtCore.Qt.Checked)       
342
343        # select 1d data
344        self.form.cbSelect.setCurrentIndex(2)
345
346        # Test the current selection
347        self.assertTrue(item1D.checkState() == QtCore.Qt.Checked)
348        self.assertTrue(item2D.checkState() == QtCore.Qt.Unchecked)       
349
350        # unselect 1d data
351        self.form.cbSelect.setCurrentIndex(3)
352
353        # Test the current selection
354        self.assertTrue(item1D.checkState() == QtCore.Qt.Unchecked)
355        self.assertTrue(item2D.checkState() == QtCore.Qt.Unchecked)       
356
357        # select 2d data
358        self.form.cbSelect.setCurrentIndex(4)
359
360        # Test the current selection
361        self.assertTrue(item1D.checkState() == QtCore.Qt.Unchecked)
362        self.assertTrue(item2D.checkState() == QtCore.Qt.Checked)       
363
364        # unselect 2d data
365        self.form.cbSelect.setCurrentIndex(5)
366
367        # Test the current selection
368        self.assertTrue(item1D.checkState() == QtCore.Qt.Unchecked)
369        self.assertTrue(item2D.checkState() == QtCore.Qt.Unchecked)       
370
371        # choose impossible index and assure the code raises
372        #with self.assertRaises(Exception):
373        #    self.form.cbSelect.setCurrentIndex(6)
374
375    def testFreezeTheory(self):
376        """
377        Assure theory freeze functionality works
378        """
379        # Not yet tested - agree on design first.
380        pass
381
382    def testRecursivelyCloneItem(self):
383        """
384        Test the rescursive QAbstractItem/QStandardItem clone
385        """
386        # Create an item with several branches
387        item1 = QtGui.QStandardItem()
388        item2 = QtGui.QStandardItem()
389        item3 = QtGui.QStandardItem()
390        item4 = QtGui.QStandardItem()
391        item5 = QtGui.QStandardItem()
392        item6 = QtGui.QStandardItem()
393
394        item4.appendRow(item5)
395        item2.appendRow(item4)
396        item2.appendRow(item6)
397        item1.appendRow(item2)
398        item1.appendRow(item3)
399
400        # Clone
401        new_item = self.form.recursivelyCloneItem(item1)
402
403        # assure the trees look identical
404        self.assertEqual(item1.rowCount(), new_item.rowCount())
405        self.assertEqual(item1.child(0).rowCount(), new_item.child(0).rowCount())
406        self.assertEqual(item1.child(1).rowCount(), new_item.child(1).rowCount())
407        self.assertEqual(item1.child(0).child(0).rowCount(), new_item.child(0).child(0).rowCount())
408
409    def testReadData(self):
410        """
411        Test the low level readData() method
412        """
413        filename = ["cyl_400_20.txt"]
414        self.form.manager.add_data = MagicMock()
415
416        # Initialize signal spy instances
417        spy_status_update = QtSignalSpy(self.form, self.form.communicator.statusBarUpdateSignal)
418        spy_data_received = QtSignalSpy(self.form, self.form.communicator.fileDataReceivedSignal)
419
420        # Read in the file
421        self.form.readData(filename)
422
423        # Expected two status bar updates
424        self.assertEqual(spy_status_update.count(), 2)
425        self.assertIn(filename[0], str(spy_status_update.called()[0]['args'][0]))
426
427
428        # Check that the model contains the item
429        self.assertEqual(self.form.model.rowCount(), 1)
430        self.assertEqual(self.form.model.columnCount(), 1)
431
432        # The 0th item header should be the name of the file
433        model_item = self.form.model.index(0,0)
434        model_name = str(self.form.model.data(model_item).toString())
435        self.assertEqual(model_name, filename[0])
436
437    def skip_testDisplayHelp(self): # Skip due to help path change
438        """
439        Test that the Help window gets shown correctly
440        """
441        partial_url = "sasgui/guiframe/data_explorer_help.html"
442        button1 = self.form.cmdHelp
443        button2 = self.form.cmdHelp_2
444
445        # Click on the Help button
446        QTest.mouseClick(button1, Qt.LeftButton)
447        QtGui.qApp.processEvents()
448
449        # Check the browser
450        self.assertIn(partial_url, str(self.form._helpView.url()))
451        # Close the browser
452        self.form._helpView.close()
453
454        # Click on the Help_2 button
455        QTest.mouseClick(button2, Qt.LeftButton)
456        QtGui.qApp.processEvents()
457        # Check the browser
458        self.assertIn(partial_url, str(self.form._helpView.url()))
459
460    def testLoadFile(self):
461        """
462        Test the threaded call to readData()
463        """
464        #self.form.loadFile()
465        pass
466
467    def testGetWList(self):
468        """
469        Test the list of known extensions
470        """
471        w_list = self.form.getWlist()
472
473        defaults = 'All (*.*);;canSAS files (*.xml);;SESANS files' +\
474            ' (*.ses);;ASCII files (*.txt);;' +\
475            'IGOR/DAT 2D Q_map files (*.dat);;IGOR 1D files (*.abs);;'+\
476            'DANSE files (*.sans)'
477        default_list = defaults.split(';;')
478
479        for def_format in default_list:
480            self.assertIn(def_format, w_list)
481       
482    def testLoadComplete(self):
483        """
484        Test the callback method updating the data object
485        """
486        message="Loading Data Complete"
487        data_dict = {"a1":Data1D()}
488        output_data = (data_dict, message)
489
490        self.form.manager.add_data = MagicMock()
491
492        # Initialize signal spy instances
493        spy_status_update = QtSignalSpy(self.form, self.form.communicator.statusBarUpdateSignal)
494        spy_data_received = QtSignalSpy(self.form, self.form.communicator.fileDataReceivedSignal)
495
496        # Read in the file
497        self.form.loadComplete(output_data)
498
499        # "Loading data complete" no longer sent in LoadFile but in callback
500        self.assertIn("Loading Data Complete", str(spy_status_update.called()[0]['args'][0]))
501
502        # Expect one Data Received signal
503        self.assertEqual(spy_data_received.count(), 1)
504
505        # Assure returned dictionary has correct data
506        # We don't know the data ID, so need to iterate over dict
507        data_dict = spy_data_received.called()[0]['args'][0]
508        for data_key, data_value in data_dict.items():
509            self.assertIsInstance(data_value, Data1D)
510
511        # Assure add_data on data_manager was called (last call)
512        self.assertTrue(self.form.manager.add_data.called)
513
514    def testNewPlot1D(self):
515        """
516        Creating new plots from Data1D/2D
517        """
518        loader = Loader()
519        manager = DataManager()
520        PlotHelper.clear()
521        self.form.enableGraphCombo(None)
522
523        # Make sure the controls are disabled
524        self.assertFalse(self.form.cbgraph.isEnabled())
525        self.assertFalse(self.form.cmdAppend.isEnabled())
526
527        # get Data1D
528        p_file="cyl_400_20.txt"
529        output_object = loader.load(p_file)
530        new_data = [manager.create_gui_data(output_object[0], p_file)]
531
532        # Mask retrieval of the data
533        self.form.plotsFromCheckedItems = MagicMock(return_value=new_data)
534
535        # Mask plotting
536        self.form.parent.workspace = MagicMock()
537
538        # Call the plotting method
539        self.form.newPlot()
540
541        # The plot was registered
542        self.assertEqual(len(PlotHelper.currentPlots()), 1)
543
544        self.assertTrue(self.form.cbgraph.isEnabled())
545        self.assertTrue(self.form.cmdAppend.isEnabled())
546
547    def testNewPlot2D(self):
548        """
549        Creating new plots from Data1D/2D
550        """
551        loader = Loader()
552        manager = DataManager()
553        PlotHelper.clear()
554        self.form.enableGraphCombo(None)
555
556        # Make sure the controls are disabled
557        self.assertFalse(self.form.cbgraph.isEnabled())
558        self.assertFalse(self.form.cmdAppend.isEnabled())
559
560        # get Data2D
561        p_file="P123_D2O_10_percent.dat"
562        output_object = loader.load(p_file)
563        new_data = [manager.create_gui_data(output_object[0], p_file)]
564
565        # Mask retrieval of the data
566        self.form.plotsFromCheckedItems = MagicMock(return_value=new_data)
567
568        # Mask plotting
569        self.form.parent.workspace = MagicMock()
570
571        # Call the plotting method
572        self.form.newPlot()
573
574        # The plot was registered
575        self.assertEqual(len(PlotHelper.currentPlots()), 1)
576
577        self.assertTrue(self.form.cbgraph.isEnabled())
578        self.assertTrue(self.form.cmdAppend.isEnabled())
579
580    @patch('sas.qtgui.Utilities.GuiUtils.plotsFromCheckedItems')
581    def testAppendPlot(self, test_patch):
582        """
583        Creating new plots from Data1D/2D
584        """
585        loader = Loader()
586        manager = DataManager()
587
588        PlotHelper.clear()
589        self.form.enableGraphCombo(None)
590
591        # Make sure the controls are disabled
592        self.assertFalse(self.form.cbgraph.isEnabled())
593        self.assertFalse(self.form.cmdAppend.isEnabled())
594
595        # get Data1D
596        p_file="cyl_400_20.txt"
597        output_object = loader.load(p_file)
598        output_item = QtGui.QStandardItem()
599        new_data = [(output_item, manager.create_gui_data(output_object[0], p_file))]
600
601        # Mask plotting
602        self.form.parent.workspace = MagicMock()
603
604        # Mask the plot show call
605        Plotter.show = MagicMock()
606
607        # Mask retrieval of the data
608        test_patch.return_value = new_data
609
610        # Call the plotting method
611        self.form.newPlot()
612
613        # Call the plotting method again, so we have 2 graphs
614        self.form.newPlot()
615
616        # See that we have two plots
617        self.assertEqual(len(PlotHelper.currentPlots()), 2)
618
619        # Add data to plot #1
620        self.form.cbgraph.setCurrentIndex(1)
621        self.form.appendPlot()
622
623        # See that we still have two plots
624        self.assertEqual(len(PlotHelper.currentPlots()), 2)
625
626    def testUpdateGraphCombo(self):
627        """
628        Test the combo box update
629        """
630        PlotHelper.clear()
631
632        graph_list=["1","2","3"]
633        self.form.updateGraphCombo(graph_list)
634
635        self.assertEqual(self.form.cbgraph.count(), 3)
636        self.assertEqual(self.form.cbgraph.currentText(), '1')
637
638        graph_list=[]
639        self.form.updateGraphCombo(graph_list)
640        self.assertEqual(self.form.cbgraph.count(), 0)
641
642    def testUpdateModelFromPerspective(self):
643        """
644        Assure the model update is correct
645        """
646        good_item = QtGui.QStandardItem()
647        bad_item = "I'm so bad"
648
649        self.form.model.reset = MagicMock()
650
651        self.form.updateModelFromPerspective(good_item)
652
653        # See that the model got reset
654        self.form.model.reset.assert_called_once()
655
656        # See that the bad item causes raise
657        with self.assertRaises(Exception):
658            self.form.updateModelFromPerspective(bad_item)
659
660    def testContextMenu(self):
661        """
662        See if the context menu is present
663        """
664        # get Data1D
665        p_file=["cyl_400_20.txt"]
666        # Read in the file
667        output, message = self.form.readData(p_file)
668        self.form.loadComplete((output, message))
669
670        # Pick up the treeview index corresponding to that file
671        index = self.form.treeView.indexAt(QtCore.QPoint(5,5))
672        self.form.show()
673
674        # Find out the center pointof the treeView row
675        rect = self.form.treeView.visualRect(index).center()
676
677        self.form.context_menu.exec_ = MagicMock()
678
679        # Move the mouse pointer to the first row
680        QTest.mouseMove(self.form.treeView.viewport(), pos=rect)
681
682        # This doesn't invoke the action/signal. Investigate why?
683        # QTest.mouseClick(self.form.treeView.viewport(), Qt.RightButton, pos=rect)
684
685        # Instead, send the signal directly
686        self.form.treeView.customContextMenuRequested.emit(rect)
687
688        # app.exec_() # debug
689
690        # See that the menu has been shown
691        self.form.context_menu.exec_.assert_called_once()
692
693    def testShowDataInfo(self):
694        """
695        Test of the showDataInfo method
696        """
697        # get Data1D
698        p_file=["cyl_400_20.txt"]
699        # Read in the file
700        output, message = self.form.readData(p_file)
701        self.form.loadComplete((output, message))
702
703        # select the data
704        self.form.treeView.selectAll()
705
706        # Call the tested method
707        self.form.showDataInfo()
708
709        # Test the properties
710        self.assertTrue(self.form.txt_widget.isReadOnly())
711        self.assertEqual(self.form.txt_widget.windowTitle(), "Data Info: cyl_400_20.txt")
712        self.assertIn("Waveln_max", self.form.txt_widget.toPlainText())
713
714        # Slider moved all the way up
715        self.assertEqual(self.form.txt_widget.verticalScrollBar().sliderPosition(), 0)
716
717    def testSaveDataAs(self):
718        """
719        Test the Save As context menu action
720        """
721        # get Data1D
722        p_file=["cyl_400_20.txt"]
723        # Read in the file
724        output, message = self.form.readData(p_file)
725        self.form.loadComplete((output, message))
726
727        # select the data
728        self.form.treeView.selectAll()
729
730        QFileDialog.getSaveFileName = MagicMock()
731
732        # Call the tested method
733        self.form.saveDataAs()
734        QFileDialog.getSaveFileName.assert_called_with(
735                                caption="Save As",
736                                directory='cyl_400_20_out.txt',
737                                filter='Text files (*.txt);;CanSAS 1D files(*.xml)',
738                                parent=None)
739        QFileDialog.getSaveFileName.assert_called_once()
740
741        # get Data2D
742        p_file=["P123_D2O_10_percent.dat"]
743        # Read in the file
744        output, message = self.form.readData(p_file)
745        self.form.loadComplete((output, message))
746
747        # select the data
748        index = self.form.model.index(1, 0)
749        selmodel = self.form.treeView.selectionModel()
750        selmodel.setCurrentIndex(index, QItemSelectionModel.NoUpdate)
751        selmodel.select(index, QItemSelectionModel.Select|QItemSelectionModel.Rows)
752
753        QFileDialog.getSaveFileName = MagicMock()
754
755        # Call the tested method
756        self.form.saveDataAs()
757        QFileDialog.getSaveFileName.assert_called_with(
758                                caption="Save As",
759                                directory='P123_D2O_10_percent_out.dat',
760                                filter='IGOR/DAT 2D file in Q_map (*.dat)',
761                                parent=None)
762        QFileDialog.getSaveFileName.assert_called_once()
763
764    def testQuickDataPlot(self):
765        """
766        Quick data plot generation.
767        """
768        # get Data1D
769        p_file=["cyl_400_20.txt"]
770        # Read in the file
771        output, message = self.form.readData(p_file)
772        self.form.loadComplete((output, message))
773
774        # select the data
775        self.form.treeView.selectAll()
776
777        Plotter.show = MagicMock() # for masking the display
778
779        self.form.quickDataPlot()
780        self.assertTrue(Plotter.show.called)
781
782    def testQuickData3DPlot(self):
783        """
784        Slow(er) 3D data plot generation.
785        """
786        # get Data1D
787        p_file=["P123_D2O_10_percent.dat"]
788        # Read in the file
789        output, message = self.form.readData(p_file)
790        self.form.loadComplete((output, message))
791
792        # select the data
793        self.form.treeView.selectAll()
794
795        Plotter2D.show = MagicMock() # for masking the display
796
797        self.form.quickData3DPlot()
798
799        self.assertTrue(Plotter2D.show.called)
800
801    def testShowEditMask(self):
802        """
803        Edit mask on a 2D plot.
804
805        TODO: add content once plotting finalized
806        """
807        pass
808
809
810if __name__ == "__main__":
811    unittest.main()
Note: See TracBrowser for help on using the repository browser.