fix: change test to match new code

Author: Gigaszi

mapswipe_workers/tests/unittests/test_spatial_sampling.py

@@ -1,29 +1,32 @@
 import os
-
 import unittest
+
 import numpy as np
 import pandas as pd
 from shapely import wkt
 from shapely.geometry import Point
 
-from mapswipe_workers.utils.spatial_sampling  import distance_on_sphere, filter_points, spatial_sampling
+from mapswipe_workers.utils.spatial_sampling import (
+    distance_on_sphere,
+    filter_points,
+    spatial_sampling,
+)
 
 
 class TestDistanceCalculations(unittest.TestCase):
-
     @classmethod
     def setUpClass(cls):
         with open(
-                os.path.join(
-                    os.path.dirname(os.path.abspath(__file__)),
-                    "..",
-                    "fixtures",
-                    "mapillary_sequence.csv",
-                ),
-                "r",
+            os.path.join(
+                os.path.dirname(os.path.abspath(__file__)),
+                "..",
+                "fixtures",
+                "mapillary_sequence.csv",
+            ),
+            "r",
         ) as file:
             df = pd.read_csv(file)
-            df['geometry'] = df['geometry'].apply(wkt.loads)
+            df["geometry"] = df["geometry"].apply(wkt.loads)
 
             cls.fixture_df = df
 
@@ -42,41 +45,43 @@ def test_filter_points(self):
                 "POINT (-74.006 40.7128)",
                 "POINT (-75.006 41.7128)",
                 "POINT (-76.006 42.7128)",
-                "POINT (-77.006 43.7128)"
+                "POINT (-77.006 43.7128)",
             ]
         }
         df = pd.DataFrame(data)
 
-        df['geometry'] = df['geometry'].apply(wkt.loads)
+        df["geometry"] = df["geometry"].apply(wkt.loads)
 
-        df['long'] = df['geometry'].apply(lambda geom: geom.x if geom.geom_type == 'Point' else None)
-        df['lat'] = df['geometry'].apply(lambda geom: geom.y if geom.geom_type == 'Point' else None)
+        df["long"] = df["geometry"].apply(
+            lambda geom: geom.x if geom.geom_type == "Point" else None
+        )
+        df["lat"] = df["geometry"].apply(
+            lambda geom: geom.y if geom.geom_type == "Point" else None
+        )
         threshold_distance = 100
         filtered_df = filter_points(df, threshold_distance)
 
         self.assertIsInstance(filtered_df, pd.DataFrame)
         self.assertLessEqual(len(filtered_df), len(df))
 
-
     def test_spatial_sampling_ordering(self):
         data = {
             "geometry": [
                 "POINT (-74.006 40.7128)",
                 "POINT (-75.006 41.7128)",
                 "POINT (-76.006 42.7128)",
-                "POINT (-77.006 43.7128)"
+                "POINT (-77.006 43.7128)",
             ],
-            'captured_at': [1, 2, 3, 4],
-            'sequence_id': ['1', '1', '1', '1']
+            "captured_at": [1, 2, 3, 4],
+            "sequence_id": ["1", "1", "1", "1"],
         }
         df = pd.DataFrame(data)
-        df['geometry'] = df['geometry'].apply(wkt.loads)
+        df["geometry"] = df["geometry"].apply(wkt.loads)
 
         interval_length = 0.1
         filtered_gdf = spatial_sampling(df, interval_length)
 
-        self.assertTrue(filtered_gdf['captured_at'].is_monotonic_increasing)
-
+        self.assertTrue(filtered_gdf["captured_at"].is_monotonic_decreasing)
 
     def test_spatial_sampling_with_sequence(self):
         threshold_distance = 0.01
@@ -86,15 +91,13 @@ def test_spatial_sampling_with_sequence(self):
 
         filtered_df.reset_index(drop=True, inplace=True)
         for i in range(len(filtered_df) - 1):
-            geom1 = filtered_df.loc[i, 'geometry']
-            geom2 = filtered_df.loc[i + 1, 'geometry']
+            geom1 = filtered_df.loc[i, "geometry"]
+            geom2 = filtered_df.loc[i + 1, "geometry"]
 
             distance = geom1.distance(geom2)
 
             self.assertLess(distance, threshold_distance)
 
 
-
-
 if __name__ == "__main__":
     unittest.main()