class FilterGeomTest(TestCase): def setUp(self): self.parser = CityParser() def test_empty_geom(self): self.assertEqual(self.parser.filter_geom('geom', None), None) self.assertFalse(self.parser.warnings) def test_invalid_geom(self): geom = MultiPolygon(Polygon(((0, 0), (0, 1), (1, 0), (1, 1), (0, 0)))) self.assertEqual(self.parser.filter_geom('geom', geom), None) self.assertTrue(self.parser.warnings) def test_polygon(self): geom = Polygon(((0, 0), (0, 1), (1, 1), (1, 0), (0, 0))) self.assertEqual(self.parser.filter_geom('geom', geom), MultiPolygon(geom)) self.assertFalse(self.parser.warnings) def test_multipolygon(self): geom = MultiPolygon(Polygon(((0, 0), (0, 1), (1, 1), (1, 0), (0, 0)))) self.assertEqual(self.parser.filter_geom('geom', geom), geom) self.assertFalse(self.parser.warnings)
def setUp(self): self.parser = CityParser()