def test_rule_count_p_n4(self): df = pd.read_csv('tests/test_files/testfile_8.csv', encoding='utf-8', delimiter=';') rule = Rule() ds = BitmapDataset(1, df) p, n = ds.count_p_n_rule(ds.unmake_rule(rule)) self.assertEqual(16, p) self.assertEqual(16, n)
def test_rule_count_p_n3(self): df = pd.read_csv('tests/test_files/testfile_8.csv', encoding='utf-8', delimiter=';') l = Literal('ClassOfSeat', 'in', '1st') rule = Rule() rule.add_literal(l) ds = BitmapDataset(1, df) p, n = ds.count_p_n_rule(ds.unmake_rule(rule)) self.assertEqual(4, p) self.assertEqual(4, n)
def test_rule_count_p_n5(self): df = pd.read_csv('tests/test_files/testfile_4.csv', encoding='utf-8', delimiter=';') l = Literal('Sex', 'in', 'Female') rule = Rule() rule.add_literal(l) ds = BitmapDataset(1, df) ds.delete_covered(ds.unmake_rule(rule)) p, n = ds.count_p_n_rule(ds.unmake_rule(rule)) self.assertEqual(0, p) self.assertEqual(0, n)