Update tests to reflect changes

datasciencecampus · Aug 8, 2023 · 69fe9a7 · 69fe9a7
1 parent eb0887c
commit 69fe9a7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/transport_performance/gtfs/routes.py b/src/transport_performance/gtfs/routes.py
@@ -98,7 +98,7 @@ def scrape_route_type_lookup(
     for url in [gtfs_url, ext_spec_url]:
         _url_defence(url)
 
-    _bool_defence(extended_schema)
+    _bool_defence(extended_schema, "extended_schema")
     # Get the basic scheme lookup
     resp_txt = _get_response_text(gtfs_url)
     soup = BeautifulSoup(resp_txt, "html.parser")

diff --git a/tests/gtfs/test_validation.py b/tests/gtfs/test_validation.py
@@ -450,10 +450,10 @@ def test_summarise_trips_on_pass(self, gtfs_fixture):
             [
                 "day",
                 "route_type",
-                "trip_count_amax",
-                "trip_count_amin",
+                "trip_count_max",
                 "trip_count_mean",
                 "trip_count_median",
+                "trip_count_min",
             ],
             dtype="object",
         )
@@ -482,8 +482,8 @@ def test_summarise_trips_on_pass(self, gtfs_fixture):
         expected_df = {
             "day": {8: "friday", 9: "friday"},
             "route_type": {8: 3, 9: 200},
-            "trip_count_amax": {8: 1211, 9: 90},
-            "trip_count_amin": {8: 1211, 9: 88},
+            "trip_count_max": {8: 1211, 9: 90},
+            "trip_count_min": {8: 1211, 9: 88},
             "trip_count_mean": {8: 1211.0, 9: 88.0},
             "trip_count_median": {8: 1211.0, 9: 88.0},
         }
@@ -519,9 +519,9 @@ def test_summarise_routes_on_pass(self, gtfs_fixture):
             [
                 "day",
                 "route_count_max",
-                "route_count_min",
                 "route_count_mean",
                 "route_count_median",
+                "route_count_min",
                 "route_type",
             ],
             dtype="object",