From 01caa92537c88359e2ed27fe8bf2e02fa010f426 Mon Sep 17 00:00:00 2001
From: samkoebrich <samkoebrich@gmail.com>
Date: Thu, 27 Feb 2025 10:05:10 -0700
Subject: [PATCH] Test for every region in my-access in maps geojson

---
 tests/test_sdk.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

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()