fix(street): adjust tests

Author: ofr1tz

mapswipe_workers/tests/unittests/test_process_mapillary.py

@@ -7,6 +7,7 @@
 from shapely import wkt
 from shapely.geometry import GeometryCollection, MultiPolygon, Point, Polygon
 
+from mapswipe_workers.definitions import CustomError
 from mapswipe_workers.utils.process_mapillary import (
     coordinate_download,
     create_tiles,
@@ -326,7 +327,7 @@ def test_get_image_metadata_no_rows(self, mock_coordinate_download):
             "start_time": "1916-01-20 00:00:00",
             "end_time": "1922-01-21 23:59:59",
         }
-        with self.assertRaises(ValueError):
+        with self.assertRaises(CustomError):
             get_image_metadata(self.fixture_data, **params)
 
     @patch("mapswipe_workers.utils.process_mapillary.coordinate_download")
@@ -335,7 +336,7 @@ def test_get_image_metadata_empty_response(self, mock_coordinate_download):
         df = df.drop(df.index)
         mock_coordinate_download.return_value = df
 
-        with self.assertRaises(ValueError):
+        with self.assertRaises(CustomError):
             get_image_metadata(self.fixture_data)
 
     @patch("mapswipe_workers.utils.process_mapillary.filter_results")
@@ -346,7 +347,7 @@ def test_get_image_metadata_size_restriction(
         mock_filter_results.return_value = pd.DataFrame({"ID": range(1, 100002)})
         mock_coordinate_download.return_value = self.fixture_df
 
-        with self.assertRaises(ValueError):
+        with self.assertRaises(CustomError):
             get_image_metadata(self.fixture_data)