diff --git a/unittests/test_geometry_factory.py b/unittests/test_geometry_factory.py index b82b70bd..910cb839 100644 --- a/unittests/test_geometry_factory.py +++ b/unittests/test_geometry_factory.py @@ -30,7 +30,7 @@ class TestGeometryFactory(TestCase): def _get_city(self, file, file_type): file_path = (self._example_path / file).resolve() - self._city = GeometryFactory(file_type, file_path).city + self._city = GeometryFactory(file_type, path=file_path).city self.assertIsNotNone(self._city, 'city is none') return self._city @@ -139,7 +139,7 @@ class TestGeometryFactory(TestCase): Test geopandas import """ file = 'sample.geojson' - city = self._get_city(file, 'geopandas') + city = self._get_city(file, 'gpandas') self.assertIsNotNone(city, 'city is none') self.assertTrue(len(city.buildings) == 1) self._check_buildings(city) @@ -169,6 +169,6 @@ class TestGeometryFactory(TestCase): :return: """ file = 'subway.osm' - city = self._get_city(file, 'geojson') + city = self._get_city(file, 'osm_subway') self.assertIsNotNone(city, 'subway entrances is none') self.assertEqual(len(city.city_objects), 20, 'Wrong number of subway entrances')