def test_grow_rule_2(self): df = pd.read_csv('tests/test_files/mushroom.csv', encoding='utf-8', delimiter=';') ds = BitmapDataset(0, df) rule = ds.grow_rule_sorted_p_n() rule = ds.make_rule(rule) self.assertEqual("a5 in ['c', 'f', 'm', 'p', 's', 'y']", rule.to_string())
def test_grow_rule_1(self): df = pd.read_csv('tests/test_files/testfile_4.csv', encoding='utf-8', delimiter=';') ds = BitmapDataset(0, df) rule = ds.grow_rule_sorted_p_n() rule = ds.make_rule(rule) self.assertEqual( "Sex in ['Female'] and Age in [4, 5, 6, 7, 35, 37, 45]", rule.to_string())