Ignore:
Timestamp:
Nov 9, 2017 8:45:20 AM (7 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:
dd150ef
Parents:
d6b8a1d
git-author:
Piotr Rozyczko <rozyczko@…> (11/08/17 09:22:45)
git-committer:
Piotr Rozyczko <rozyczko@…> (11/09/17 08:45:20)
Message:

Converted unit tests

File:
1 edited

Legend:

Unmodified
Added
Removed

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

    r7fb471d r53c771e  
    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                return True 
    3940            def setData(self, data_item=None, is_batch=False): 
    4041                return None 
    … …  
    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 
    … …  
    180181        item2 = self.form.model.item(1) 
    181182        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) 
     183        self.assertTrue(item1.checkState() == Qt.Checked) 
     184        self.assertTrue(item2.checkState() == Qt.Checked) 
     185        self.assertTrue(item3.checkState() == Qt.Checked) 
    185186 
    186187        # Click on the delete  button 
    … …  
    188189 
    189190        # Test the warning dialog called once 
    190         self.assertTrue(QtGui.QMessageBox.question.called) 
     191        self.assertTrue(QMessageBox.question.called) 
    191192 
    192193        # Assure the model still contains the items 
    … …  
    194195 
    195196        # Now, mock the confirmation dialog with return=Yes 
    196         QtGui.QMessageBox.question = MagicMock(return_value=QtGui.QMessageBox.Yes) 
     197        QMessageBox.question = MagicMock(return_value=QMessageBox.Yes) 
    197198 
    198199        # Click on the delete  button 
    … …  
    200201 
    201202        # Test the warning dialog called once 
    202         self.assertTrue(QtGui.QMessageBox.question.called) 
     203        self.assertTrue(QMessageBox.question.called) 
    203204 
    204205        # Assure the model contains no items 
    … …  
    215216 
    216217        # Mock the confirmation dialog with return=No 
    217         QtGui.QMessageBox.question = MagicMock(return_value=QtGui.QMessageBox.No) 
     218        QMessageBox.question = MagicMock(return_value=QMessageBox.No) 
    218219 
    219220        # Populate the model 
    220         item1 = QtGui.QStandardItem(True) 
     221        item1 = QStandardItem(True) 
    221222        item1.setCheckable(True) 
    222         item1.setCheckState(QtCore.Qt.Checked) 
     223        item1.setCheckState(Qt.Checked) 
    223224        item1.setText("item 1") 
    224225        self.form.theory_model.appendRow(item1) 
    225         item2 = QtGui.QStandardItem(True) 
     226        item2 = QStandardItem(True) 
    226227        item2.setCheckable(True) 
    227         item2.setCheckState(QtCore.Qt.Unchecked) 
     228        item2.setCheckState(Qt.Unchecked) 
    228229        item2.setText("item 2") 
    229230        self.form.theory_model.appendRow(item2) 
    … …  
    233234 
    234235        # Assure the checkboxes are on 
    235         self.assertTrue(item1.checkState() == QtCore.Qt.Checked) 
    236         self.assertTrue(item2.checkState() == QtCore.Qt.Unchecked) 
     236        self.assertTrue(item1.checkState() == Qt.Checked) 
     237        self.assertTrue(item2.checkState() == Qt.Unchecked) 
    237238 
    238239        # Click on the delete  button 
    … …  
    240241 
    241242        # Test the warning dialog called once 
    242         self.assertTrue(QtGui.QMessageBox.question.called) 
     243        self.assertTrue(QMessageBox.question.called) 
    243244 
    244245        # Assure the model still contains the items 
    … …  
    246247 
    247248        # Now, mock the confirmation dialog with return=Yes 
    248         QtGui.QMessageBox.question = MagicMock(return_value=QtGui.QMessageBox.Yes) 
     249        QMessageBox.question = MagicMock(return_value=QMessageBox.Yes) 
    249250 
    250251        # Click on the delete  button 
    … …  
    252253 
    253254        # Test the warning dialog called once 
    254         self.assertTrue(QtGui.QMessageBox.question.called) 
     255        self.assertTrue(QMessageBox.question.called) 
    255256 
    256257        # Assure the model contains 1 item 
    … …  
    258259 
    259260        # Set the remaining item to checked 
    260         self.form.theory_model.item(0).setCheckState(QtCore.Qt.Checked) 
     261        self.form.theory_model.item(0).setCheckState(Qt.Checked) 
    261262 
    262263        # Click on the delete button again 
    … …  
    306307 
    307308        # Mock the warning message 
    308         QtGui.QMessageBox = MagicMock() 
     309        QMessageBox = MagicMock() 
    309310 
    310311        # Click on the button 
    … …  
    312313 
    313314        # Assure the message box popped up 
    314         QtGui.QMessageBox.assert_called_once() 
     315        QMessageBox.assert_called_once() 
    315316 
    316317    def testDataSelection(self): 
    … …  
    325326        self.form.cbSelect.setCurrentIndex(1) 
    326327 
    327         self.form.show() 
    328         app.exec_() 
    329  
    330328        # Test the current selection 
    331329        item1D = self.form.model.item(0) 
    332330        item2D = self.form.model.item(1) 
    333         self.assertTrue(item1D.checkState() == QtCore.Qt.Unchecked) 
    334         self.assertTrue(item2D.checkState() == QtCore.Qt.Unchecked)         
     331        self.assertTrue(item1D.checkState() == Qt.Unchecked) 
     332        self.assertTrue(item2D.checkState() == Qt.Unchecked)         
    335333 
    336334        # Select all data 
    … …  
    338336 
    339337        # Test the current selection 
    340         self.assertTrue(item1D.checkState() == QtCore.Qt.Checked) 
    341         self.assertTrue(item2D.checkState() == QtCore.Qt.Checked)         
     338        self.assertTrue(item1D.checkState() == Qt.Checked) 
     339        self.assertTrue(item2D.checkState() == Qt.Checked)         
    342340 
    343341        # select 1d data 
    … …  
    345343 
    346344        # Test the current selection 
    347         self.assertTrue(item1D.checkState() == QtCore.Qt.Checked) 
    348         self.assertTrue(item2D.checkState() == QtCore.Qt.Unchecked)         
     345        self.assertTrue(item1D.checkState() == Qt.Checked) 
     346        self.assertTrue(item2D.checkState() == Qt.Unchecked)         
    349347 
    350348        # unselect 1d data 
    … …  
    352350 
    353351        # Test the current selection 
    354         self.assertTrue(item1D.checkState() == QtCore.Qt.Unchecked) 
    355         self.assertTrue(item2D.checkState() == QtCore.Qt.Unchecked)         
     352        self.assertTrue(item1D.checkState() == Qt.Unchecked) 
     353        self.assertTrue(item2D.checkState() == Qt.Unchecked)         
    356354 
    357355        # select 2d data 
    … …  
    359357 
    360358        # Test the current selection 
    361         self.assertTrue(item1D.checkState() == QtCore.Qt.Unchecked) 
    362         self.assertTrue(item2D.checkState() == QtCore.Qt.Checked)         
     359        self.assertTrue(item1D.checkState() == Qt.Unchecked) 
     360        self.assertTrue(item2D.checkState() == Qt.Checked)         
    363361 
    364362        # unselect 2d data 
    … …  
    366364 
    367365        # Test the current selection 
    368         self.assertTrue(item1D.checkState() == QtCore.Qt.Unchecked) 
    369         self.assertTrue(item2D.checkState() == QtCore.Qt.Unchecked)         
     366        self.assertTrue(item1D.checkState() == Qt.Unchecked) 
     367        self.assertTrue(item2D.checkState() == Qt.Unchecked)         
    370368 
    371369        # choose impossible index and assure the code raises 
    … …  
    385383        """ 
    386384        # 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() 
     385        item1 = QStandardItem() 
     386        item2 = QStandardItem() 
     387        item3 = QStandardItem() 
     388        item4 = QStandardItem() 
     389        item5 = QStandardItem() 
     390        item6 = QStandardItem() 
    393391 
    394392        item4.appendRow(item5) 
    … …  
    432430        # The 0th item header should be the name of the file 
    433431        model_item = self.form.model.index(0,0) 
    434         model_name = str(self.form.model.data(model_item).toString()) 
     432        model_name = self.form.model.data(model_item) 
    435433        self.assertEqual(model_name, filename[0]) 
    436434 
    … …  
    445443        # Click on the Help button 
    446444        QTest.mouseClick(button1, Qt.LeftButton) 
    447         QtGui.qApp.processEvents() 
     445        qApp.processEvents() 
    448446 
    449447        # Check the browser 
    … …  
    454452        # Click on the Help_2 button 
    455453        QTest.mouseClick(button2, Qt.LeftButton) 
    456         QtGui.qApp.processEvents() 
     454        qApp.processEvents() 
    457455        # Check the browser 
    458456        self.assertIn(partial_url, str(self.form._helpView.url())) 
    … …  
    596594        p_file="cyl_400_20.txt" 
    597595        output_object = loader.load(p_file) 
    598         output_item = QtGui.QStandardItem() 
     596        output_item = QStandardItem() 
    599597        new_data = [(output_item, manager.create_gui_data(output_object[0], p_file))] 
    600598 
    … …  
    644642        Assure the model update is correct 
    645643        """ 
    646         good_item = QtGui.QStandardItem() 
     644        good_item = QStandardItem() 
    647645        bad_item = "I'm so bad" 
    648646 
    … …  
    669667 
    670668        # Pick up the treeview index corresponding to that file 
    671         index = self.form.treeView.indexAt(QtCore.QPoint(5,5)) 
     669        index = self.form.treeView.indexAt(QPoint(5,5)) 
    672670        self.form.show() 
    673671 
Note: See TracChangeset for help on using the changeset viewer.