1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
# Test methods with long descriptive names can omit docstrings
# pylint: disable=missing-docstring
import unittest
from AnyQt.QtCore import Qt
from Orange.data import Table
from Orange.widgets.model.owrandomforest import OWRandomForest
from Orange.widgets.tests.base import (
WidgetTest,
DefaultParameterMapping,
ParameterMapping,
WidgetLearnerTestMixin
)
class TestOWRandomForest(WidgetTest, WidgetLearnerTestMixin):
def setUp(self):
self.widget = self.create_widget(
OWRandomForest, stored_settings={"auto_apply": False})
self.init()
nest_spin = self.widget.n_estimators_spin
nest_min_max = [nest_spin.minimum() * 10, nest_spin.minimum()]
self.parameters = [
ParameterMapping("n_estimators", nest_spin, nest_min_max),
ParameterMapping("min_samples_split",
self.widget.min_samples_split_spin[1])]
def test_parameters_checked(self):
"""Check learner and model for various values of all parameters
when all properties are checked
"""
self.widget.max_features_spin[0].setCheckState(Qt.Checked)
self.widget.max_depth_spin[0].setCheckState(Qt.Checked)
self.parameters.extend([
ParameterMapping("max_features", self.widget.max_features_spin[1]),
ParameterMapping("max_depth", self.widget.max_depth_spin[1])])
self.test_parameters()
def test_parameters_unchecked(self):
"""Check learner and model for various values of all parameters
when properties are not checked
"""
self.widget.min_samples_split_spin[0].setCheckState(Qt.Unchecked)
self.parameters = self.parameters[:1]
self.parameters.extend([
DefaultParameterMapping("max_features", "sqrt"),
DefaultParameterMapping("random_state", None),
DefaultParameterMapping("max_depth", None),
DefaultParameterMapping("min_samples_split", 2)])
self.test_parameters()
def test_class_weights(self):
table = Table("iris")
self.send_signal(self.widget.Inputs.data, table)
self.assertFalse(self.widget.class_weight)
self.widget.controls.class_weight.setChecked(True)
self.assertTrue(self.widget.class_weight)
self.click_apply()
self.assertEqual(self.widget.model.skl_model.class_weight, "balanced")
self.assertTrue(self.widget.Warning.class_weights_used.is_shown())
if __name__ == "__main__":
unittest.main()
|
