Refactor test_api.py to remove unnecessary characters from file path

philsv · Oct 21, 2024 · afbb1e5 · afbb1e5
1 parent a2c6e68
commit afbb1e5
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/...hly_['countryRegionId', 'productId'].json → ...55_monthly_countryRegionId_productId.json b/...hly_['countryRegionId', 'productId'].json → ...55_monthly_countryRegionId_productId.json
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -98,6 +98,16 @@ def test_get_series_via_route(route, series, frequency, facet, mocker):
     file_path = get_mock_data_path(
         f"{route.replace('/', '-')}_{series}_{frequency}_{facet}.json"
     )
+
+    file_path = (
+        file_path.replace("[", "")
+        .replace("]", "")
+        .replace("'", "")
+        .replace(" ", "")
+        .replace(",", "_")
+        .strip()
+    )
+
     spy_get = mock_requests_get(mocker, file_path)
 
     df = eia.get_series_via_route(route, series, frequency, facet)