def test_make_rule_2(self):
df = pd.read_csv('tests/test_files/mushroom.csv',
encoding='utf-8',
delimiter=';')
ds = BitmapDataset(0, df)
rule = list()
rule = ds.make_rule(rule)
self.assertEqual("", rule.to_string())
def test_grow_rule_4(self):
df = pd.read_csv('tests/test_files/mushroom.csv',
encoding='utf-8',
delimiter=';')
ds = BitmapDataset(0, df)
rule = ds.grow_rule_inductive()
rule = ds.make_rule(rule)
self.assertEqual("a5 in ['c', 'f', 'm', 'p', 's', 'y']",
rule.to_string())
def test_grow_rule_3(self):
df = pd.read_csv('tests/test_files/testfile_4.csv',
encoding='utf-8',
delimiter=';')
ds = BitmapDataset(0, df)
rule = ds.grow_rule_inductive()
rule = ds.make_rule(rule)
self.assertEqual("Sex in ['Female'] and Age in [4, 5, 6, 7, 8, 9]",
rule.to_string())
def test_make_rule_1(self):
df = pd.read_csv('tests/test_files/mushroom.csv',
encoding='utf-8',
delimiter=';')
ds = BitmapDataset(0, df)
rule = list()
rule.append([0, 0])
rule.append([2, 3])
rule.append([5, 1])
rule.append([5, 0])
rule = ds.make_rule(rule)
self.assertEqual("a1 in ['x'] and a3 in ['g'] and a6 in ['a', 'f']",
rule.to_string())
def test_prune_rule_2(self):
df = pd.read_csv('tests/test_files/mushroom.csv',
encoding='utf-8',
delimiter=';')
ds = BitmapDataset(0, df)
l = Literal('a5', 'in', ['c', 'f', 'm', 'p', 's', 'y'])
rule = Rule()
rule.add_literal(l)
rule = ds.unmake_rule(rule)
rule = ds.prune_rule(rule)
rule = ds.make_rule(rule)
self.assertEqual("a5 in ['c', 'f', 'm', 'p', 's', 'y']",
rule.to_string())
def test_prune_rule_1(self):
df = pd.read_csv('tests/test_files/testfile_3.csv',
encoding='utf-8',
delimiter=';')
l = Literal('Sex', 'in', 'Female')
l2 = Literal('Sex', 'in', 'Male')
rule = Rule()
rule.add_literal(l)
rule.add_literal(l2)
ds = BitmapDataset(1, df)
rule = ds.prune_rule(ds.unmake_rule(rule))
rule = ds.make_rule(rule)
self.assertEqual("", rule.to_string())
