class TestCorrelationRank(WidgetTest):
@classmethod
def setUpClass(cls):
super().setUpClass()
cls.iris = Table("iris")
cls.attrs = cls.iris.domain.attributes
def setUp(self):
self.vizrank = CorrelationRank(None)
self.vizrank.attrs = self.attrs
def test_compute_score(self):
self.vizrank.master = Mock()
self.vizrank.master.cont_data = self.iris
self.vizrank.master.correlation_type = CorrelationType.PEARSON
npt.assert_almost_equal(self.vizrank.compute_score((1, 0)),
[-0.1094, -0.1094, 0.1828], 4)
def test_row_for_state(self):
row = self.vizrank.row_for_state((-0.2, 0.2, 0.1), (1, 0))
self.assertEqual(row[0].data(Qt.DisplayRole), "+0.200")
self.assertEqual(row[0].data(CorrelationRank.PValRole), 0.1)
self.assertEqual(row[1].data(Qt.DisplayRole), self.attrs[0].name)
self.assertEqual(row[2].data(Qt.DisplayRole), self.attrs[1].name)
def test_iterate_states_by_feature(self):
self.vizrank.sel_feature_index = 2
states = self.vizrank.iterate_states_by_feature()
self.assertListEqual([(2, 0), (2, 1), (2, 3)], list(states))
class TestCorrelationRank(WidgetTest):
@classmethod
def setUpClass(cls):
super().setUpClass()
cls.iris = Table("iris")
cls.attrs = cls.iris.domain.attributes
def setUp(self):
self.vizrank = CorrelationRank(None)
self.vizrank.attrs = self.attrs
def test_compute_score(self):
self.vizrank.master = Mock()
self.vizrank.master.cont_data = self.iris
self.vizrank.master.correlation_type = CorrelationType.PEARSON
npt.assert_almost_equal(self.vizrank.compute_score((1, 0)),
[-0.1094, -0.1094, 0.1828], 4)
def test_row_for_state(self):
row = self.vizrank.row_for_state((-0.2, 0.2, 0.1), (1, 0))
self.assertEqual(row[0].data(Qt.DisplayRole), "+0.200")
self.assertEqual(row[0].data(CorrelationRank.PValRole), 0.1)
self.assertEqual(row[1].data(Qt.DisplayRole), self.attrs[0].name)
self.assertEqual(row[2].data(Qt.DisplayRole), self.attrs[1].name)
def test_iterate_states(self):
self.assertListEqual(list(self.vizrank.iterate_states(None)),
[(1, 0), (2, 0), (2, 1), (3, 0), (3, 1), (3, 2)])
self.assertListEqual(list(self.vizrank.iterate_states((1, 0))),
[(1, 0), (2, 0), (2, 1), (3, 0), (3, 1), (3, 2)])
self.assertListEqual(list(self.vizrank.iterate_states((2, 1))),
[(2, 1), (3, 0), (3, 1), (3, 2)])
def test_iterate_states_by_feature(self):
self.vizrank.sel_feature_index = 2
states = self.vizrank.iterate_states_by_feature()
self.assertListEqual([(2, 0), (2, 1), (2, 3)], list(states))
def test_state_count(self):
self.assertEqual(self.vizrank.state_count(), 6)
self.vizrank.sel_feature_index = 2
self.assertEqual(self.vizrank.state_count(), 3)
def setUp(self):
self.vizrank = CorrelationRank(None)
self.vizrank.attrs = self.attrs
def setUp(self):
self.vizrank = CorrelationRank(None)
self.vizrank.attrs = self.attrs
