import sys import unittest from bumps import options from PyQt4 import QtGui from PyQt4 import QtWebKit from mock import MagicMock # set up import paths import path_prepare from UnitTesting.TestUtils import QtSignalSpy # Local from sas.qtgui.Perspectives.Fitting.FittingOptions import FittingOptions if not QtGui.QApplication.instance(): app = QtGui.QApplication(sys.argv) class FittingOptionsTest(unittest.TestCase): '''Test the FittingOptions dialog''' def setUp(self): '''Create FittingOptions dialog''' self.widget = FittingOptions(None, config=options.FIT_CONFIG) def tearDown(self): '''Destroy the GUI''' self.widget.close() self.widget = None def testDefaults(self): '''Test the GUI in its default state''' self.assertIsInstance(self.widget, QtGui.QDialog) # Default title self.assertEqual(self.widget.windowTitle(), "Fit Algorithms") # The combo box self.assertIsInstance(self.widget.cbAlgorithm, QtGui.QComboBox) self.assertEqual(self.widget.cbAlgorithm.count(), 5) self.assertEqual(self.widget.cbAlgorithm.itemText(0), 'Nelder-Mead Simplex') self.assertEqual(self.widget.cbAlgorithm.itemText(4), 'Levenberg-Marquardt') self.assertEqual(self.widget.cbAlgorithm.currentIndex(), 4) def testAssignValidators(self): """ Check that line edits got correct validators """ # Can't reliably test the method in action, but can easily check the results # DREAM self.assertIsInstance(self.widget.samples_dream.validator(), QtGui.QIntValidator) self.assertIsInstance(self.widget.burn_dream.validator(), QtGui.QIntValidator) self.assertIsInstance(self.widget.pop_dream.validator(), QtGui.QDoubleValidator) self.assertIsInstance(self.widget.thin_dream.validator(), QtGui.QIntValidator) self.assertIsInstance(self.widget.steps_dream.validator(), QtGui.QIntValidator) # DE self.assertIsInstance(self.widget.steps_de.validator(), QtGui.QIntValidator) self.assertIsInstance(self.widget.CR_de.validator(), QtGui.QDoubleValidator) self.assertIsInstance(self.widget.pop_de.validator(), QtGui.QDoubleValidator) self.assertIsInstance(self.widget.F_de.validator(), QtGui.QDoubleValidator) self.assertIsInstance(self.widget.ftol_de.validator(), QtGui.QDoubleValidator) self.assertIsInstance(self.widget.xtol_de.validator(), QtGui.QDoubleValidator) # bottom value for floats and ints self.assertEqual(self.widget.steps_de.validator().bottom(), 0) self.assertEqual(self.widget.CR_de.validator().bottom(), 0) # Behaviour on empty cell self.widget.onAlgorithmChange(3) self.widget.steps_de.setText("") # This should disable the OK button ## self.assertFalse(self.widget.buttonBox.button(QtGui.QDialogButtonBox.Ok).isEnabled()) # Let's put some valid value in lineedit self.widget.steps_de.setText("1") # This should enable the OK button self.assertTrue(self.widget.buttonBox.button(QtGui.QDialogButtonBox.Ok).isEnabled()) def testOnAlgorithmChange(self): '''Test the combo box change callback''' # Current ID self.assertEqual(self.widget.current_fitter_id, 'lm') # index = 0 self.widget.onAlgorithmChange(0) # Check Nelder-Mead self.assertEqual(self.widget.stackedWidget.currentIndex(), 1) self.assertEqual(self.widget.current_fitter_id, 'lm') # index = 4 self.widget.onAlgorithmChange(4) # Check Levenberg-Marquad self.assertEqual(self.widget.stackedWidget.currentIndex(), 1) self.assertEqual(self.widget.current_fitter_id, 'lm') def testOnApply(self): '''Test bumps update''' # Spy on the update signal spy_apply = QtSignalSpy(self.widget, self.widget.fit_option_changed) # Set the DREAM optimizer self.widget.cbAlgorithm.setCurrentIndex(2) # Change some values self.widget.init_dream.setCurrentIndex(2) self.widget.steps_dream.setText("50") # Apply the new values self.widget.onApply() self.assertEqual(spy_apply.count(), 1) self.assertIn('DREAM', spy_apply.called()[0]['args'][0]) # Check the parameters self.assertEqual(options.FIT_CONFIG.values['dream']['steps'], 50.0) self.assertEqual(options.FIT_CONFIG.values['dream']['init'], 'cov') def testOnHelp(self): ''' Test help display''' #Mock the QWebView method QtWebKit.QWebView.show = MagicMock() QtWebKit.QWebView.load = MagicMock() # Invoke the action on default tab self.widget.onHelp() # Check if show() got called self.assertTrue(QtWebKit.QWebView.show.called) # Assure the filename is correct self.assertIn("optimizer.html", QtWebKit.QWebView.load.call_args[0][0].toString()) # Change the combo index self.widget.cbAlgorithm.setCurrentIndex(2) self.widget.onHelp() # Check if show() got called self.assertEqual(QtWebKit.QWebView.show.call_count, 2) # Assure the filename is correct self.assertIn("fit-dream", QtWebKit.QWebView.load.call_args[0][0].toString()) # Change the index again self.widget.cbAlgorithm.setCurrentIndex(4) self.widget.onHelp() # Check if show() got called self.assertEqual(QtWebKit.QWebView.show.call_count, 3) # Assure the filename is correct self.assertIn("fit-lm", QtWebKit.QWebView.load.call_args[0][0].toString()) def testWidgetFromOptions(self): '''Test the helper function''' # test empty call self.assertIsNone(self.widget.widgetFromOption(None)) # test silly call self.assertIsNone(self.widget.widgetFromOption('poop')) self.assertIsNone(self.widget.widgetFromOption(QtGui.QMainWindow())) # Switch to DREAM self.widget.cbAlgorithm.setCurrentIndex(2) # test smart call self.assertIsInstance(self.widget.widgetFromOption('samples'), QtGui.QLineEdit) self.assertIsInstance(self.widget.widgetFromOption('init'), QtGui.QComboBox) def testUpdateWidgetFromBumps(self): '''Test the widget update''' # modify some value options.FIT_CONFIG.values['newton']['steps'] = 1234 options.FIT_CONFIG.values['newton']['starts'] = 666 options.FIT_CONFIG.values['newton']['xtol'] = 0.01 # Invoke the method for the changed self.widget.updateWidgetFromBumps('newton') # See that the widget picked up the right values self.assertEqual(self.widget.steps_newton.text(), '1234') self.assertEqual(self.widget.starts_newton.text(), '666') self.assertEqual(self.widget.ftol_newton.text(), '1e-06') # default self.assertEqual(self.widget.xtol_newton.text(), '0.01') if __name__ == "__main__": unittest.main()