Col and index filtering test

ImperialCollegeLondon · Sep 25, 2023 · 7c6f706 · 7c6f706
1 parent fc0bdcf
commit 7c6f706
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/test_dsr_api.py b/tests/test_dsr_api.py
@@ -100,3 +100,10 @@ def test_get_dsr_api(dsr_data):
     assert len(response.json()["data"][0].keys()) == 2
     assert "Activity Types" in response.json()["data"][0].keys()
     assert "kWh Cost" in response.json()["data"][0].keys()
+
+    response = client.get("/dsr?start=1&col=activities")
+    assert len(response.json()["data"]) == 2
+    assert len(response.json()["data"][0].keys()) == 1
+    assert "Activities" in response.json()["data"][0].keys()
+    assert len(response.json()["data"][1].keys()) == 1
+    assert "Activities" in response.json()["data"][1].keys()