def testGetCC(self): index = country_code_converter.create_index(table) row = country_code_converter.get_cc('name_english', 'Austria', index=index) rowdict = dict([(country_code_converter.columns[i], val) for (i, val) in enumerate(table[0])]) self.assertEqual(rowdict, row) col = country_code_converter.get_cc('name_english', 'Austria', 'car', index=index) self.assertEqual(table[0][5], col)
def testOutput(self): self.assertEqual(country_code_converter.get_cc('name_english', 'Italy', 'iso2'), 'IT') self.assertEqual(country_code_converter.get_cc('name_german', 'Deutschland', 'car'), 'D') self.assertEqual(country_code_converter.get_cc('iso3', 'AUT', 'centroid'), (47.333333, 13.333333))