def __init__(self, countries_map: Map, palette): super(Visualizer, self).__init__() countries_map.calc_incident_countries() self.colorer = Colorer(countries_map, palette) self.countries_to_polygons = self.update_map() self.reset_action = None self.highlighted_country = None self.initUI()
class TestMap(TestCase): def setUp(self): self.c1 = Country.from_json([[[0, 0], [200, 200], [100, 300]]]) self.c2 = Country.from_json([[[0, 0], [200, 0], [200, 200]]]) self.c3 = Country.from_json([[[0, 0], [0, 300], [100, 300]]]) self.c4 = Country.from_json([[[200, 200], [100, 300], [500, 500]]]) self.c_map = Map([self.c1, self.c2, self.c3, self.c4]) def test_calc_incident_countries(self): self.c_map.calc_incident_countries() self.assertTrue(self.c2.is_neighbour(self.c1)) self.assertTrue(self.c4.is_neighbour(self.c1)) def test_set_sizes(self): self.assertTrue(self.c_map.width, 500) self.assertTrue(self.c_map.height, 500)