def test_find_unique_classes(self): columns = Task2.import_columns('sub_table.csv') observed = Task2.find_unique_classes(columns['class']) self.assertEqual(observed, {'Mammalia', 'Reptilia', 'Aves'})