source: sasview/src/sas/qtgui/UnitTesting/DataExplorerTest.py @ 39551a68

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 39551a68 was 39551a68, checked in by Piotr Rozyczko <piotr.rozyczko@…>, 8 years ago

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