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