source: sasview/src/sas/qtgui/Calculators/UnitTesting/ResolutionCalculatorPanelTest.py @ 5aad7a5

ESS_GUIESS_GUI_batch_fittingESS_GUI_bumps_abstractionESS_GUI_iss1116ESS_GUI_iss879ESS_GUI_openclESS_GUI_orderingESS_GUI_sync_sascalc
Last change on this file since 5aad7a5 was 144fe21, checked in by Piotr Rozyczko <rozyczko@…>, 6 years ago

new unit test runner script + test fixes. SASVIEW-970
  • Property mode set to 100644
File size: 12.3 KB
RevLine 
[01cda57]1import sys
2import time
3import numpy
4import logging
5import unittest
[53c771e]6from PyQt5 import QtGui, QtWidgets
7from PyQt5 import QtCore
8from PyQt5.QtTest import QTest
9from PyQt5.QtCore import Qt
[7fb471d]10from unittest.mock import MagicMock
[01cda57]11
12from twisted.internet import threads
13
[144fe21]14from sas.qtgui.Utilities.GuiUtils import Communicate
[01cda57]15from sas.qtgui.Calculators.ResolutionCalculatorPanel \
16    import ResolutionCalculatorPanel
17
18BG_COLOR_ERR = 'background-color: rgb(244, 170, 164);'
19
20BG_COLOR_WARNING = 'background-color: rgb(244, 217, 164);'
21
22
[53c771e]23if not QtWidgets.QApplication.instance():
24    app = QtWidgets.QApplication(sys.argv)
[01cda57]25
26
27class ResolutionCalculatorPanelTest(unittest.TestCase):
28    """Test the ResolutionCalculator"""
29    def setUp(self):
[144fe21]30        class dummy_manager(object):
31            communicator = Communicate()
[01cda57]32        """Create the ResolutionCalculator"""
[144fe21]33        self.widget = ResolutionCalculatorPanel(dummy_manager())
[01cda57]34
35        # self.widget.onCompute = MagicMock()
36
37    def tearDown(self):
38        """Destroy the ResolutionCalculator"""
[144fe21]39        if self.widget.plotter is not None:
40            self.widget.plotter.close()
[01cda57]41        self.widget.close()
42        self.widget = None
43
44    def testDefaults(self):
45        """Test the GUI in its default state"""
46
[53c771e]47        self.assertIsInstance(self.widget, QtWidgets.QDialog)
[01cda57]48        self.assertEqual(self.widget.windowTitle(), "Q Resolution Estimator")
49        # size
[170e95d]50        self.assertEqual(self.widget.size().height(), 540)
51        self.assertEqual(self.widget.size().width(), 876)
[01cda57]52
53        # visibility
54        self.assertFalse(self.widget.lblSpectrum.isVisible())
55        self.assertFalse(self.widget.cbCustomSpectrum.isVisible())
56
57        # content of line edits
58        self.assertEqual(self.widget.txtDetectorPixSize.text(), '0.5, 0.5')
59        self.assertEqual(self.widget.txtDetectorSize.text(), '128, 128')
60        self.assertEqual(self.widget.txtSample2DetectorDistance.text(), '1000')
61        self.assertEqual(self.widget.txtSampleApertureSize.text(), '1.27')
62        self.assertEqual(self.widget.txtSampleOffset.text(), '0')
63        self.assertEqual(self.widget.txtSource2SampleDistance.text(), '1627')
64        self.assertEqual(self.widget.txtSourceApertureSize.text(), '3.81')
65        self.assertEqual(self.widget.txtWavelength.text(), '6.0')
66        self.assertEqual(self.widget.txtWavelengthSpread.text(), '0.125')
67
68        self.assertEqual(self.widget.txtQx.text(), '0.0')
69        self.assertEqual(self.widget.txtQy.text(), '0.0')
70
71        self.assertEqual(self.widget.txt1DSigma.text(), '0.0008289')
72        self.assertEqual(self.widget.txtSigma_lamd.text(), '3.168e-05')
73        self.assertEqual(self.widget.txtSigma_x.text(), '0.0008288')
74        self.assertEqual(self.widget.txtSigma_x.text(), '0.0008288')
75
76        # items of comboboxes
77        self.assertFalse(self.widget.cbCustomSpectrum.isEditable())
78        self.assertEqual(self.widget.cbCustomSpectrum.currentText(), 'Flat')
79        self.assertEqual(self.widget.cbCustomSpectrum.currentIndex(), 0)
80        self.assertListEqual([self.widget.cbCustomSpectrum.itemText(i) for i in
81                              range(self.widget.cbCustomSpectrum.count())],
82                             ['Flat', 'Add New'])
83
84        self.assertFalse(self.widget.cbSource.isEditable())
85        self.assertEqual(self.widget.cbSource.count(), 6)
86        self.assertEqual(self.widget.cbSource.currentText(), 'Neutron')
87        self.assertListEqual([self.widget.cbSource.itemText(i) for i in
88                              range(self.widget.cbSource.count())],
89                             ['Alpha', 'Deutron', 'Neutron', 'Photon',
90                              'Proton', 'Triton'])
91
92        self.assertFalse(self.widget.cbWaveColor.isEditable())
93        self.assertEqual(self.widget.cbWaveColor.count(), 2)
94        self.assertEqual(self.widget.cbWaveColor.currentText(), 'Monochromatic')
95        self.assertListEqual([self.widget.cbWaveColor.itemText(i) for i in
96                              range(self.widget.cbWaveColor.count())],
97                             ['Monochromatic', 'TOF'])
98
99        # read only text edits
100        self.assertTrue(self.widget.txtSigma_x.isReadOnly())
101        self.assertTrue(self.widget.txtSigma_y.isReadOnly())
102        self.assertTrue(self.widget.txtSigma_lamd.isReadOnly())
103        self.assertTrue(self.widget.txt1DSigma.isReadOnly())
104
105        # tooltips
106        self.assertEqual(self.widget.cbSource.toolTip(),
107                         "Source Selection: "
108                         "Affect on the gravitational contribution.")
109        self.widget.cbCustomSpectrum.toolTip(), \
110        "Wavelength Spectrum: Intensity vs. wavelength."
111        # print self.widget.txtDetectorPixSize.toolTip()
112        self.assertEqual(self.widget.txtDetectorPixSize.toolTip(),
113                         "Detector Pixel Size.")
114        self.assertEqual(self.widget.txtDetectorSize.toolTip(),
115                         "Number of Pixels on Detector.")
116        self.assertEqual(self.widget.txtSample2DetectorDistance.toolTip(),
117                         "Sample Aperture to Detector Distance.")
118        self.assertEqual(self.widget.txtSampleApertureSize.toolTip(),
119                         "Sample Aperture Size.")
120        self.assertEqual(self.widget.txtSampleOffset.toolTip(),
121                         "Sample Offset.")
122        self.assertEqual(self.widget.txtSource2SampleDistance.toolTip(),
123                         "Source to Sample Aperture Distance.")
124        self.assertEqual(self.widget.txtSourceApertureSize.toolTip(),
125                         "Source Aperture Size.")
126        self.assertEqual(self.widget.txtWavelength.toolTip(),
127                         "Wavelength of the Neutrons.")
128        self.assertEqual(self.widget.txtWavelengthSpread.toolTip(),
129                         "Wavelength Spread of Neutrons.")
130        self.assertEqual(self.widget.txtQx.toolTip(), "Type the Qx value.")
131        self.assertEqual(self.widget.txtQy.toolTip(), "Type the Qy value.")
132        self.assertEqual(self.widget.txt1DSigma.toolTip(),
133                         "Resolution in 1-dimension (for 1D data).")
134        self.assertEqual(self.widget.txtSigma_lamd.toolTip(),
135                         "The wavelength contribution in the radial direction."
136                         " Note: The phi component is always zero.")
137        self.assertEqual(self.widget.txtSigma_x.toolTip(),
138                         "The x component of the geometric resolution, "
139                         "excluding sigma_lamda.")
140        self.assertEqual(self.widget.txtSigma_y.toolTip(),
141                         "The y component of the geometric resolution, "
142                         "excluding sigma_lamda.")
143
144    def testFormatNumber(self):
145        self.assertEqual(self.widget.formatNumber('  7.123456  '), '7.123')
146
147    def testCheckWavelength(self):
148        """ Test validator for Wavelength text edit"""
149        self.widget.txtWavelength.clear()
150        # Enter invalid input for Monochromatic spectrum
151        # check that background becomes red and Compute button is disabled
152        QTest.keyClicks(self.widget.txtWavelength, 'vcv ')
153        QTest.keyClick(self.widget.txtWavelength, QtCore.Qt.Key_Return)
154        self.assertIn(BG_COLOR_ERR, self.widget.txtWavelength.styleSheet())
155        self.assertFalse(self.widget.cmdCompute.isEnabled())
156
157        # Enter invalid input for TOF spectrum
158        # check that background becomes red and Compute button is disabled
159        self.widget.cbWaveColor.setCurrentIndex(1)
160
161        self.widget.txtWavelength.clear()
162        QTest.keyClicks(self.widget.txtWavelength, '4')
163        QTest.keyClick(self.widget.txtWavelength, QtCore.Qt.Key_Return)
164        self.assertIn(BG_COLOR_ERR, self.widget.txtWavelength.styleSheet())
165        self.assertFalse(self.widget.cmdCompute.isEnabled())
166
167    def testCheckWavelengthSpread(self):
168        """ Test validator for WavelengthSpread """
169        self.widget.txtWavelengthSpread.clear()
170        QTest.keyClicks(self.widget.txtWavelengthSpread, '0.12; 1.3')
171        QTest.keyClick(self.widget.txtWavelengthSpread, QtCore.Qt.Key_Return)
172        self.assertIn(BG_COLOR_ERR,
173                      self.widget.txtWavelengthSpread.styleSheet())
174
175    def testCheckPixels(self):
176        """ Test validator for pixel size and number """
177        self.widget.txtDetectorPixSize.clear()
178        QTest.keyClicks(self.widget.txtDetectorPixSize, '0.12; 1.3')
179        QTest.keyClick(self.widget.txtDetectorPixSize, QtCore.Qt.Key_Return)
180        self.assertIn(BG_COLOR_ERR,
181                      self.widget.txtDetectorPixSize.styleSheet())
182
183        self.widget.txtDetectorSize.clear()
184        QTest.keyClicks(self.widget.txtDetectorSize, '0.12')
185        QTest.keyClick(self.widget.txtDetectorSize, QtCore.Qt.Key_Return)
186        self.assertIn(BG_COLOR_ERR,
187                      self.widget.txtDetectorSize.styleSheet())
188
189    def testCheckQx_y(self):
190        """ Test validator for qx and qy inputs """
191        self.widget.txtQx.clear()
192        QTest.keyClicks(self.widget.txtQx, '0.12; 1.3')
193        QTest.keyClick(self.widget.txtQx, QtCore.Qt.Key_Return)
194        self.assertIn(BG_COLOR_ERR,
195                      self.widget.txtQx.styleSheet())
196        # put back default value
197        self.widget.txtQx.setText('0.0')
198
199        self.widget.txtQy.clear()
200        QTest.keyClicks(self.widget.txtQy, '0.12, a')
201        QTest.keyClick(self.widget.txtQy, QtCore.Qt.Key_Return)
202        self.assertIn(BG_COLOR_ERR, self.widget.txtQy.styleSheet())
203        # put back default value
204        self.widget.txtQy.setText('0.0')
205
206    def testOnSelectWaveColor(self):
207        """ Test change of layout if type of source is TOF """
208        # choose TOF
209        AllItems = [self.widget.cbWaveColor.itemText(i)
210                    for i in range(self.widget.cbWaveColor.count())]
211        self.widget.cbWaveColor.setCurrentIndex(AllItems.index('TOF'))
212
213        # call function
214        self.widget.onSelectWaveColor()
215        self.widget.show()
216
217        # check that TOF is selected
218        self.assertTrue(self.widget.cbWaveColor.currentText(), 'TOF')
219
220        # check modifications of Wavelength text edit: min - max
221        self.assertEqual(self.widget.txtWavelength.text(), '6.0 - 12.0')
222        self.assertEqual(self.widget.txtWavelengthSpread.text(), '0.125 - 0.125')
223
224        # check that Spectrum label and cbCustomSpectrum are visible
225        self.assertTrue(self.widget.lblSpectrum.isVisible())
226        self.assertTrue(self.widget.cbCustomSpectrum.isVisible())
227
228    def testOnSelectCustomSpectrum(self):
229        """ Test Custom Spectrum: load file if 'Add New' """
[53c771e]230        QtWidgets.QFileDialog.getOpenFileName = MagicMock(return_value=("",""))
[01cda57]231        self.widget.cbCustomSpectrum.setCurrentIndex(1)
232
233        # Test the getOpenFileName() dialog called once
[53c771e]234        self.assertTrue(QtWidgets.QFileDialog.getOpenFileName.called)
235        QtWidgets.QFileDialog.getOpenFileName.assert_called_once()
[01cda57]236
237    def testHelp(self):
238        """ Assure help file is shown """
239        # this should not rise
[e90988c]240        self.widget.manager = QtWidgets.QWidget()
241        self.widget.manager.showHelp = MagicMock()
[01cda57]242        self.widget.onHelp()
[e90988c]243        self.assertTrue(self.widget.manager.showHelp.called_once())
244        args = self.widget.manager.showHelp.call_args
245        self.assertIn('resolution_calculator_help.html', args[0][0])
[01cda57]246
247    def testOnReset(self):
248        """ Test onReset function"""
249        # modify gui
250        self.widget.txtQx.setText('33.0')
251        self.widget.cbSource.setCurrentIndex(0)
252
253        # check that GUI has been modified
254        self.assertNotEqual(self.widget.cbSource.currentText(), 'Neutron')
255        # apply reset
256        QTest.mouseClick(self.widget.cmdReset, Qt.LeftButton)
257        # check that we get back to the initial state
258        self.assertEqual(self.widget.txtQx.text(), '0.0')
259        self.assertEqual(self.widget.cbSource.currentText(), 'Neutron')
260
261    def testOnClose(self):
262        """ test Closing window """
263        closeButton = self.widget.cmdClose
264        QTest.mouseClick(closeButton, Qt.LeftButton)
265
266    def testOnCompute(self):
267        """ """
268        threads.deferToThread = MagicMock()
269        self.widget.onCompute()
270
271        # thread called
272        self.assertTrue(threads.deferToThread.called)
[fc4fec8]273        self.assertEqual(threads.deferToThread.call_args_list[0][0][0].__name__, 'map_wrapper')
[01cda57]274
275        # the Compute button changed caption and got disabled
276        self.assertEqual(self.widget.cmdCompute.text(), 'Wait...')
277        self.assertFalse(self.widget.cmdCompute.isEnabled())
278
279
280if __name__ == "__main__":
281    unittest.main()
Note: See TracBrowser for help on using the repository browser.