def test_find_best_char_literal_3(self): df = pd.read_csv('tests/test_files/testfile_3.csv', encoding='utf-8', delimiter=';') ds = DictDataset(1, df) best_l, best_foil = ds.find_best_literal(0, 0, df['Sex'].unique(), 'Sex') self.assertEqual(9.38889, round(best_foil, 5)) self.assertEqual("Sex in ['Female']", best_l.to_string())
def test_find_best_num_literal_3(self): df = pd.read_csv('tests/test_files/testfile_3.csv', encoding='utf-8', delimiter=';') ds = DictDataset(1, df) best_l, best_foil = ds.find_best_literal(0, 0, df['Age'].unique(), 'Age') self.assertEqual(10, round(best_foil, 5)) self.assertEqual('Age > 88', best_l.to_string())
def test_find_best_char_literal_2(self): df = pd.read_csv('tests/test_files/testfile_2.csv', encoding='utf-8', delimiter=';') ds = DictDataset(1, df) best_l, best_foil = ds.find_best_literal(0, 0, df['ClassOfSeat'].unique(), 'ClassOfSeat') self.assertEqual(9, round(best_foil, 5)) self.assertEqual("ClassOfSeat in ['1st', '3rd', 'Crew']", best_l.to_string())