source: sasview/src/sas/qtgui/UnitTesting/DataExplorerTest.py @ 64f1e93

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 64f1e93 was 6d05e1d, checked in by Piotr Rozyczko <rozyczko@…>, 7 years ago

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