-
-
Notifications
You must be signed in to change notification settings - Fork 1k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
2 changed files
with
108 additions
and
95 deletions.
There are no files selected for viewing
150 changes: 56 additions & 94 deletions
150
Orange/widgets/classify/tests/test_owsvmclassification.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,107 +1,69 @@ | ||
| # Test methods with long descriptive names can omit docstrings | ||
| # pylint: disable=missing-docstring | ||
| from PyQt4 import QtGui | ||
|
|
||
| from Orange.data import Table | ||
| from Orange.widgets.classify.owsvmclassification import OWSVMClassification | ||
| from Orange.widgets.tests.base import WidgetTest | ||
| from Orange.widgets.tests.base import (WidgetTest, DefaultParameterMapping, | ||
| ParameterMapping, WidgetLearnerTestMixin) | ||
|
|
||
|
|
||
| class TestOWSVMClassification(WidgetTest): | ||
| class TestOWSVMClassification(WidgetTest, WidgetLearnerTestMixin): | ||
| def setUp(self): | ||
| self.widget = self.create_widget(OWSVMClassification) | ||
| self.widget.spin_boxes = self.widget.findChildren(QtGui.QDoubleSpinBox) | ||
| # max iter spin | ||
| self.widget.spin_boxes.append(self.widget.findChildren(QtGui.QSpinBox)[0]) | ||
| # max iter checkbox | ||
| self.widget.max_iter_check_box = self.widget.findChildren(QtGui.QCheckBox)[0] | ||
| self.spin_boxes = self.widget.spin_boxes | ||
| self.event_data = None | ||
|
|
||
| def test_kernel_equation_run(self): | ||
| """ Check if right text is written for specific kernel """ | ||
| for i in range(0, 4): | ||
| if self.widget.kernel_box.buttons[i].isChecked(): | ||
| self.assertEqual(self.widget.kernel_eq, self.widget.kernels[i][1]) | ||
|
|
||
| def test_kernel_equation(self): | ||
| """ Check if right text is written for specific kernel after click """ | ||
| for index in range(0, 4): | ||
| self.widget.kernel_box.buttons[index].click() | ||
| self.assertEqual(self.widget.kernel_eq, self.widget.kernels[index][1]) | ||
| self.widget = self.create_widget(OWSVMClassification, | ||
| stored_settings={"auto_apply": False}) | ||
| self.init() | ||
| gamma_spin = self.widget._kernel_params[0] | ||
| values = [self.widget._default_gamma, gamma_spin.maximum()] | ||
|
|
||
| def test_kernel_display_run(self): | ||
| """ Check if right spinner box for selected kernel are visible after widget start """ | ||
| for button_pos, value in ((0, [False, False, False]), | ||
| (1, [True, True, True]), | ||
| (2, [True, False, False]), | ||
| (3, [True, True, False])): | ||
| if self.widget.kernel_box.buttons[button_pos].isChecked(): | ||
| self.assertEqual([not self.spin_boxes[i].box.isHidden() for i in range(2, 5)], | ||
| value) | ||
| break | ||
| def getter(): | ||
| value = gamma_spin.value() | ||
| return gamma_spin.specialValueText() \ | ||
| if value == gamma_spin.minimum() else value | ||
|
|
||
| def test_kernel_display(self): | ||
| """ Check if right spinner box for selected kernel are visible after we select kernel """ | ||
| for button_pos, value in ((0, [False, False, False]), | ||
| (1, [True, True, True]), | ||
| (2, [True, False, False]), | ||
| (3, [True, True, False])): | ||
| self.widget.kernel_box.buttons[button_pos].click() | ||
| self.widget.kernel_box.buttons[button_pos].isChecked() | ||
| self.assertEqual([not self.spin_boxes[i].box.isHidden() for i in range(2, 5)], value) | ||
| def setter(value): | ||
| if value == gamma_spin.specialValueText(): | ||
| gamma_spin.setValue(gamma_spin.minimum()) | ||
| else: | ||
| gamma_spin.setValue(value) | ||
|
|
||
| def test_optimization_box_visible(self): | ||
| """ Check if both spinner box is visible after starting widget """ | ||
| self.assertEqual(self.spin_boxes[5].box.isHidden(), False) | ||
| self.assertEqual(self.spin_boxes[6].box.isHidden(), False) | ||
| self.parameters = [ | ||
| ParameterMapping("C", self.widget.c_spin), | ||
| ParameterMapping("gamma", self.widget._kernel_params[0], | ||
| values=values, setter=setter, getter=getter), | ||
| ParameterMapping("coef0", self.widget._kernel_params[1]), | ||
| ParameterMapping("degree", self.widget._kernel_params[2]), | ||
| ParameterMapping("tol", self.widget.tol_spin), | ||
| ParameterMapping("max_iter", self.widget.max_iter_spin[1])] | ||
|
|
||
| def test_optimization_box_checked(self): | ||
| """ Check if spinner box for iteration limit is enabled or disabled """ | ||
| for value in (True, False): | ||
| self.widget.max_iter_check_box.setChecked(value) | ||
| self.assertEqual(self.widget.max_iter_check_box.isChecked(), value) | ||
| self.assertEqual(self.spin_boxes[6].isEnabled(), value) | ||
| def test_parameters_unchecked(self): | ||
| """Check learner and model for various values of all parameters | ||
| when Iteration limit is not checked | ||
| """ | ||
| self.widget.max_iter_spin[0].setCheckState(False) | ||
| self.parameters[-1] = DefaultParameterMapping("max_iter", -1) | ||
| self.test_parameters() | ||
|
|
||
| def test_type_button_checked(self): | ||
| """ Check if SVM type is selected after click """ | ||
| self.widget.type_box.buttons[0].click() | ||
| self.assertEqual(self.widget.type_box.buttons[0].isChecked(), True) | ||
| self.widget.type_box.buttons[1].click() | ||
| self.assertEqual(self.widget.type_box.buttons[1].isChecked(), True) | ||
| def test_parameters_svm_type(self): | ||
| """Check learner and model for various values of all parameters | ||
| when NuSVM is chosen | ||
| """ | ||
| self.assertEqual(self.widget.svmtype, OWSVMClassification.C_SVC) | ||
| # setChecked(True) does not trigger callback event | ||
| self.widget.nu_radio.click() | ||
| self.assertEqual(self.widget.svmtype, OWSVMClassification.Nu_SVC) | ||
| self.parameters[0] = ParameterMapping("nu", self.widget.nu_spin) | ||
| self.test_parameters() | ||
|
|
||
| def test_type_button_properties_visible(self): | ||
| """ Check if spinner box in SVM type are visible """ | ||
| self.assertEqual(not self.spin_boxes[0].isHidden(), True) | ||
| self.assertEqual(not self.spin_boxes[1].isHidden(), True) | ||
|
|
||
| def test_data_before_apply(self): | ||
| """ Check if data are set """ | ||
| self.widget.set_data(Table("iris")[:100]) | ||
| self.widget.apply() | ||
| self.assertEqual(len(self.widget.data), 100) | ||
|
|
||
| def test_output_signal_learner(self): | ||
| """ Check if we have on output learner """ | ||
| self.widget.kernel_box.buttons[0].click() | ||
| self.widget.set_data(Table("iris")[:100]) | ||
| self.widget.apply() | ||
| self.assertNotEqual(self.widget.learner, None) | ||
| def test_kernel_equation(self): | ||
| """Check if the right equation is written according to kernel """ | ||
| for i in range(4): | ||
| self.widget.kernel_box.buttons[i].click() | ||
| self.assertEqual(self.widget.kernel_eq, self.widget.kernels[i][1]) | ||
|
|
||
| def test_output_params(self): | ||
| """ Check ouput params """ | ||
| self.widget.kernel_box.buttons[0].click() | ||
| self.widget.set_data(Table("iris")[:100]) | ||
| self.widget.max_iter_check_box.setChecked(True) | ||
| self.widget.apply() | ||
| self.widget.type_box.buttons[0].click() | ||
| params = self.widget.learner.params | ||
| self.assertEqual(params.get('C'), self.spin_boxes[0].value()) | ||
| self.widget.type_box.buttons[1].click() | ||
| params = self.widget.learner.params | ||
| self.assertEqual(params.get('nu'), self.spin_boxes[1].value()) | ||
| self.assertEqual(params.get('gamma'), self.spin_boxes[2].value()) | ||
| self.assertEqual(params.get('coef0'), self.spin_boxes[3].value()) | ||
| self.assertEqual(params.get('degree'), self.spin_boxes[4].value()) | ||
| self.assertEqual(params.get('tol'), self.spin_boxes[5].value()) | ||
| self.assertEqual(params.get('max_iter'), self.spin_boxes[6].value()) | ||
| def test_kernel_spins(self): | ||
| """Check if the right spins are visible according to kernel """ | ||
| for i, hidden in enumerate([[True, True, True], | ||
| [False, False, False], | ||
| [False, True, True], | ||
| [False, False, True]]): | ||
| self.widget.kernel_box.buttons[i].click() | ||
| self.assertEqual([self.widget._kernel_params[j].box.isHidden() | ||
| for j in range(3)], hidden) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,11 +1,62 @@ | ||
| # Test methods with long descriptive names can omit docstrings | ||
| # pylint: disable=missing-docstring | ||
| from Orange.widgets.regression.owsvmregression import OWSVMRegression | ||
| from Orange.widgets.tests.base import WidgetTest, WidgetLearnerTestMixin | ||
| from Orange.widgets.tests.base import (WidgetTest, WidgetLearnerTestMixin, | ||
| ParameterMapping) | ||
|
|
||
|
|
||
| class TestOWSVMRegression(WidgetTest, WidgetLearnerTestMixin): | ||
| def setUp(self): | ||
| self.widget = self.create_widget(OWSVMRegression, | ||
| stored_settings={"auto_apply": False}) | ||
| self.init() | ||
| gamma_spin = self.widget._kernel_params[0] | ||
| values = [self.widget._default_gamma, gamma_spin.maximum()] | ||
|
|
||
| def getter(): | ||
| value = gamma_spin.value() | ||
| return gamma_spin.specialValueText() \ | ||
| if value == gamma_spin.minimum() else value | ||
|
|
||
| def setter(value): | ||
| if value == gamma_spin.specialValueText(): | ||
| gamma_spin.setValue(gamma_spin.minimum()) | ||
| else: | ||
| gamma_spin.setValue(value) | ||
|
|
||
| self.parameters = [ | ||
| ParameterMapping("C", self.widget.epsilon_C_spin), | ||
| ParameterMapping("epsilon", self.widget.epsilon_spin), | ||
| ParameterMapping("gamma", self.widget._kernel_params[0], | ||
| values=values, setter=setter, getter=getter), | ||
| ParameterMapping("coef0", self.widget._kernel_params[1]), | ||
| ParameterMapping("degree", self.widget._kernel_params[2]), | ||
| ParameterMapping("tol", self.widget.tol_spin)] | ||
|
|
||
| def test_parameters_svr_type(self): | ||
| """Check learner and model for various values of all parameters | ||
| when NuSVR is chosen | ||
| """ | ||
| self.assertEqual(self.widget.svrtype, OWSVMRegression.Epsilon_SVR) | ||
| # setChecked(True) does not trigger callback event | ||
| self.widget.nu_radio.click() | ||
| self.assertEqual(self.widget.svrtype, OWSVMRegression.Nu_SVR) | ||
| self.parameters[0] = ParameterMapping("C", self.widget.nu_C_spin) | ||
| self.parameters[1] = ParameterMapping("nu", self.widget.nu_spin) | ||
| self.test_parameters() | ||
|
|
||
| def test_kernel_equation(self): | ||
| """Check if the right equation is written according to kernel """ | ||
| for i in range(4): | ||
| self.widget.kernel_box.buttons[i].click() | ||
| self.assertEqual(self.widget.kernel_eq, self.widget.kernels[i][1]) | ||
|
|
||
| def test_kernel_spins(self): | ||
| """Check if the right spins are visible according to kernel """ | ||
| for i, hidden in enumerate([[True, True, True], | ||
| [False, False, False], | ||
| [False, True, True], | ||
| [False, False, True]]): | ||
| self.widget.kernel_box.buttons[i].click() | ||
| self.assertEqual([self.widget._kernel_params[j].box.isHidden() | ||
| for j in range(3)], hidden) |