Ignore:
Timestamp:
Nov 8, 2017 7:22:45 AM (6 years ago)
Author:
Piotr Rozyczko <rozyczko@…>
Branches:
ESS_GUI, ESS_GUI_Docs, ESS_GUI_batch_fitting, ESS_GUI_bumps_abstraction, ESS_GUI_iss1116, ESS_GUI_iss879, ESS_GUI_iss959, ESS_GUI_opencl, ESS_GUI_ordering, ESS_GUI_sync_sascalc
Children:
0e80658
Parents:
412e069e
Message:

Converted unit tests

File:
1 edited

Legend:

Unmodified
Added
Removed

  • src/sas/qtgui/MainWindow/UnitTesting/DataExplorerTest.py

    r99ea1b0 r3e8dee3  
    22import unittest 
    33 
    4 from PyQt4.QtGui import * 
    5 from PyQt4.QtTest import QTest 
    6 from PyQt4.QtCore import * 
     4from PyQt5.QtGui import * 
     5from PyQt5.QtWidgets import * 
     6from PyQt5.QtTest import QTest 
     7from PyQt5.QtCore import * 
    78from unittest.mock import MagicMock 
    89from unittest.mock import patch 
    … …  
    3637                return Communicate() 
    3738            def allowBatch(self): 
    38                 return False 
    39             def setData(self, data_item=None): 
     39                return True 
     40            def setData(self, data_item=None, is_batch=False): 
    4041                return None 
    4142            def title(self): 
    … …  
    114115 
    115116        # Return no files. 
    116         QtGui.QFileDialog.getOpenFileNames = MagicMock(return_value=None) 
     117        QFileDialog.getOpenFileNames = MagicMock(return_value=('','')) 
    117118 
    118119        # Click on the Load button 
    … …  
    120121 
    121122        # Test the getOpenFileName() dialog called once 
    122         self.assertTrue(QtGui.QFileDialog.getOpenFileNames.called) 
    123         QtGui.QFileDialog.getOpenFileNames.assert_called_once() 
     123        self.assertTrue(QFileDialog.getOpenFileNames.called) 
     124        QFileDialog.getOpenFileNames.assert_called_once() 
    124125 
    125126        # Make sure the signal has not been emitted 
    … …  
    127128 
    128129        # Now, return a single file 
    129         QtGui.QFileDialog.getOpenFileNames = MagicMock(return_value=filename) 
     130        QFileDialog.getOpenFileNames = MagicMock(return_value=(filename,'')) 
    130131 
    131132        # Click on the Load button 
    132133        QTest.mouseClick(loadButton, Qt.LeftButton) 
    133         QtGui.qApp.processEvents() 
     134        qApp.processEvents() 
    134135 
    135136        # Test the getOpenFileName() dialog called once 
    136         self.assertTrue(QtGui.QFileDialog.getOpenFileNames.called) 
    137         QtGui.QFileDialog.getOpenFileNames.assert_called_once() 
     137        self.assertTrue(QFileDialog.getOpenFileNames.called) 
     138        QFileDialog.getOpenFileNames.assert_called_once() 
    138139 
    139140        # Expected one spy instance 
    … …  
    167168 
    168169        # Mock the confirmation dialog with return=No 
    169         QtGui.QMessageBox.question = MagicMock(return_value=QtGui.QMessageBox.No) 
     170        QMessageBox.question = MagicMock(return_value=QMessageBox.No) 
    170171 
    171172        # Populate the model 
    … …  
    181182        item2 = self.form.model.item(1) 
    182183        item3 = self.form.model.item(2) 
    183         self.assertTrue(item1.checkState() == QtCore.Qt.Checked) 
    184         self.assertTrue(item2.checkState() == QtCore.Qt.Checked) 
    185         self.assertTrue(item3.checkState() == QtCore.Qt.Checked) 
     184        self.assertTrue(item1.checkState() == Qt.Checked) 
     185        self.assertTrue(item2.checkState() == Qt.Checked) 
     186        self.assertTrue(item3.checkState() == Qt.Checked) 
    186187 
    187188        # Click on the delete  button 
    … …  
    189190 
    190191        # Test the warning dialog called once 
    191         self.assertTrue(QtGui.QMessageBox.question.called) 
     192        self.assertTrue(QMessageBox.question.called) 
    192193 
    193194        # Assure the model still contains the items 
    … …  
    195196 
    196197        # Now, mock the confirmation dialog with return=Yes 
    197         QtGui.QMessageBox.question = MagicMock(return_value=QtGui.QMessageBox.Yes) 
     198        QMessageBox.question = MagicMock(return_value=QMessageBox.Yes) 
    198199 
    199200        # Click on the delete  button 
    … …  
    201202 
    202203        # Test the warning dialog called once 
    203         self.assertTrue(QtGui.QMessageBox.question.called) 
     204        self.assertTrue(QMessageBox.question.called) 
    204205 
    205206        # Assure the model contains no items 
    … …  
    216217 
    217218        # Mock the confirmation dialog with return=No 
    218         QtGui.QMessageBox.question = MagicMock(return_value=QtGui.QMessageBox.No) 
     219        QMessageBox.question = MagicMock(return_value=QMessageBox.No) 
    219220 
    220221        # Populate the model 
    221         item1 = QtGui.QStandardItem(True) 
     222        item1 = QStandardItem(True) 
    222223        item1.setCheckable(True) 
    223         item1.setCheckState(QtCore.Qt.Checked) 
     224        item1.setCheckState(Qt.Checked) 
    224225        item1.setText("item 1") 
    225226        self.form.theory_model.appendRow(item1) 
    226         item2 = QtGui.QStandardItem(True) 
     227        item2 = QStandardItem(True) 
    227228        item2.setCheckable(True) 
    228         item2.setCheckState(QtCore.Qt.Unchecked) 
     229        item2.setCheckState(Qt.Unchecked) 
    229230        item2.setText("item 2") 
    230231        self.form.theory_model.appendRow(item2) 
    … …  
    234235 
    235236        # Assure the checkboxes are on 
    236         self.assertTrue(item1.checkState() == QtCore.Qt.Checked) 
    237         self.assertTrue(item2.checkState() == QtCore.Qt.Unchecked) 
     237        self.assertTrue(item1.checkState() == Qt.Checked) 
     238        self.assertTrue(item2.checkState() == Qt.Unchecked) 
    238239 
    239240        # Click on the delete  button 
    … …  
    241242 
    242243        # Test the warning dialog called once 
    243         self.assertTrue(QtGui.QMessageBox.question.called) 
     244        self.assertTrue(QMessageBox.question.called) 
    244245 
    245246        # Assure the model still contains the items 
    … …  
    247248 
    248249        # Now, mock the confirmation dialog with return=Yes 
    249         QtGui.QMessageBox.question = MagicMock(return_value=QtGui.QMessageBox.Yes) 
     250        QMessageBox.question = MagicMock(return_value=QMessageBox.Yes) 
    250251 
    251252        # Click on the delete  button 
    … …  
    253254 
    254255        # Test the warning dialog called once 
    255         self.assertTrue(QtGui.QMessageBox.question.called) 
     256        self.assertTrue(QMessageBox.question.called) 
    256257 
    257258        # Assure the model contains 1 item 
    … …  
    259260 
    260261        # Set the remaining item to checked 
    261         self.form.theory_model.item(0).setCheckState(QtCore.Qt.Checked) 
     262        self.form.theory_model.item(0).setCheckState(Qt.Checked) 
    262263 
    263264        # Click on the delete button again 
    … …  
    307308 
    308309        # Mock the warning message 
    309         QtGui.QMessageBox = MagicMock() 
     310        QMessageBox = MagicMock() 
    310311 
    311312        # Click on the button 
    … …  
    313314 
    314315        # Assure the message box popped up 
    315         QtGui.QMessageBox.assert_called_once() 
     316        QMessageBox.assert_called_once() 
    316317 
    317318    def testDataSelection(self): 
    … …  
    326327        self.form.cbSelect.setCurrentIndex(1) 
    327328 
    328         self.form.show() 
    329         app.exec_() 
    330  
    331329        # Test the current selection 
    332330        item1D = self.form.model.item(0) 
    333331        item2D = self.form.model.item(1) 
    334         self.assertTrue(item1D.checkState() == QtCore.Qt.Unchecked) 
    335         self.assertTrue(item2D.checkState() == QtCore.Qt.Unchecked)         
     332        self.assertTrue(item1D.checkState() == Qt.Unchecked) 
     333        self.assertTrue(item2D.checkState() == Qt.Unchecked)         
    336334 
    337335        # Select all data 
    … …  
    339337 
    340338        # Test the current selection 
    341         self.assertTrue(item1D.checkState() == QtCore.Qt.Checked) 
    342         self.assertTrue(item2D.checkState() == QtCore.Qt.Checked)         
     339        self.assertTrue(item1D.checkState() == Qt.Checked) 
     340        self.assertTrue(item2D.checkState() == Qt.Checked)         
    343341 
    344342        # select 1d data 
    … …  
    346344 
    347345        # Test the current selection 
    348         self.assertTrue(item1D.checkState() == QtCore.Qt.Checked) 
    349         self.assertTrue(item2D.checkState() == QtCore.Qt.Unchecked)         
     346        self.assertTrue(item1D.checkState() == Qt.Checked) 
     347        self.assertTrue(item2D.checkState() == Qt.Unchecked)         
    350348 
    351349        # unselect 1d data 
    … …  
    353351 
    354352        # Test the current selection 
    355         self.assertTrue(item1D.checkState() == QtCore.Qt.Unchecked) 
    356         self.assertTrue(item2D.checkState() == QtCore.Qt.Unchecked)         
     353        self.assertTrue(item1D.checkState() == Qt.Unchecked) 
     354        self.assertTrue(item2D.checkState() == Qt.Unchecked)         
    357355 
    358356        # select 2d data 
    … …  
    360358 
    361359        # Test the current selection 
    362         self.assertTrue(item1D.checkState() == QtCore.Qt.Unchecked) 
    363         self.assertTrue(item2D.checkState() == QtCore.Qt.Checked)         
     360        self.assertTrue(item1D.checkState() == Qt.Unchecked) 
     361        self.assertTrue(item2D.checkState() == Qt.Checked)         
    364362 
    365363        # unselect 2d data 
    … …  
    367365 
    368366        # Test the current selection 
    369         self.assertTrue(item1D.checkState() == QtCore.Qt.Unchecked) 
    370         self.assertTrue(item2D.checkState() == QtCore.Qt.Unchecked)         
     367        self.assertTrue(item1D.checkState() == Qt.Unchecked) 
     368        self.assertTrue(item2D.checkState() == Qt.Unchecked)         
    371369 
    372370        # choose impossible index and assure the code raises 
    … …  
    386384        """ 
    387385        # Create an item with several branches 
    388         item1 = QtGui.QStandardItem() 
    389         item2 = QtGui.QStandardItem() 
    390         item3 = QtGui.QStandardItem() 
    391         item4 = QtGui.QStandardItem() 
    392         item5 = QtGui.QStandardItem() 
    393         item6 = QtGui.QStandardItem() 
     386        item1 = QStandardItem() 
     387        item2 = QStandardItem() 
     388        item3 = QStandardItem() 
     389        item4 = QStandardItem() 
     390        item5 = QStandardItem() 
     391        item6 = QStandardItem() 
    394392 
    395393        item4.appendRow(item5) 
    … …  
    433431        # The 0th item header should be the name of the file 
    434432        model_item = self.form.model.index(0,0) 
    435         model_name = str(self.form.model.data(model_item).toString()) 
     433        model_name = self.form.model.data(model_item) 
    436434        self.assertEqual(model_name, filename[0]) 
    437435 
    … …  
    446444        # Click on the Help button 
    447445        QTest.mouseClick(button1, Qt.LeftButton) 
    448         QtGui.qApp.processEvents() 
     446        qApp.processEvents() 
    449447 
    450448        # Check the browser 
    … …  
    455453        # Click on the Help_2 button 
    456454        QTest.mouseClick(button2, Qt.LeftButton) 
    457         QtGui.qApp.processEvents() 
     455        qApp.processEvents() 
    458456        # Check the browser 
    459457        self.assertIn(partial_url, str(self.form._helpView.url())) 
    … …  
    597595        p_file="cyl_400_20.txt" 
    598596        output_object = loader.load(p_file) 
    599         output_item = QtGui.QStandardItem() 
     597        output_item = QStandardItem() 
    600598        new_data = [(output_item, manager.create_gui_data(output_object[0], p_file))] 
    601599 
    … …  
    645643        Assure the model update is correct 
    646644        """ 
    647         good_item = QtGui.QStandardItem() 
     645        good_item = QStandardItem() 
    648646        bad_item = "I'm so bad" 
    649647 
    … …  
    670668 
    671669        # Pick up the treeview index corresponding to that file 
    672         index = self.form.treeView.indexAt(QtCore.QPoint(5,5)) 
     670        index = self.form.treeView.indexAt(QPoint(5,5)) 
    673671        self.form.show() 
    674672 
Note: See TracChangeset for help on using the changeset viewer.