test: Refactor tests dependent on PKG_PATH

datasciencecampus · Jul 3, 2024 · 084fbef · 084fbef
1 parent 9670407
commit 084fbef
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/test_routes.py b/tests/test_routes.py
@@ -134,7 +134,7 @@ def test_lookup_is_stable(self):
         """Check if the tables at the urls have changed content."""
         # import the expected fixtures
         lookup_fix = pd.read_pickle(
-            os.path.join(PKG_PATH, "data", "gtfs", "route_lookup.pkl")
+            os.path.join(PKG_PATH, "data", "route_lookup.pkl")
         )
         lookup = scrape_route_type_lookup()
         pd.testing.assert_frame_equal(lookup, lookup_fix)

diff --git a/tests/test_validation.py b/tests/test_validation.py
@@ -108,7 +108,7 @@ def test_init_on_pass(self):
         with_pth = GtfsInstance(
             gtfs_pth=GTFS_FIX_PTH,
             route_lookup_pth=(
-                os.path.join(PKG_PATH, "data", "gtfs", "route_lookup.pkl")
+                os.path.join(PKG_PATH, "data", "route_lookup.pkl")
             ),
         ).ROUTE_LKP
         assert (