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)