diff --git a/tests/test_sdk.py b/tests/test_sdk.py index e3df9e4c..0441d1e3 100644 --- a/tests/test_sdk.py +++ b/tests/test_sdk.py @@ -405,6 +405,7 @@ def test_horizon_hours(self): class TestWattTimeMaps(unittest.TestCase): def setUp(self): self.maps = WattTimeMaps() + self.myaccess = WattTimeMyAccess() def test_get_maps_json_moer(self): moer = self.maps.get_maps_json(signal_type="co2_moer") @@ -441,6 +442,22 @@ def test_region_from_loc(self): self.assertEqual(region["region_full_name"], "Public Service Co of Colorado") self.assertEqual(region["signal_type"], "co2_moer") + def test_my_access_in_geojson(self): + access = self.myaccess.get_access_pandas() + for signal_type in ["co2_moer", "co2_aoer", "health_damage"]: + access_regions = access.loc[ + access["signal_type"] == signal_type, "region" + ].unique() + maps = self.maps.get_maps_json(signal_type=signal_type) + maps_regions = [i["properties"]["region"] for i in maps["features"]] + + assert ( + set(access_regions) - set(maps_regions) == set() + ), f"Missing regions in geojson for {signal_type}: {set(access_regions) - set(maps_regions)}" + assert ( + set(maps_regions) - set(access_regions) == set() + ), f"Extra regions in geojson for {signal_type}: {set(maps_regions) - set(access_regions)}" + if __name__ == "__main__": unittest.main()