def test_zones_from_geometries_exist(self):
world_geometries = json.load(open("web/geo/world.geojson"))
world_geometries_zone_keys = set()
for ft in world_geometries["features"]:
world_geometries_zone_keys.add(ft["properties"]["zoneName"])
all_zone_keys = set(ZONES_CONFIG.keys())
non_existing_zone_keys = sorted(world_geometries_zone_keys - all_zone_keys)
assert (
len(non_existing_zone_keys) == 0
), f"{non_existing_zone_keys} are defined in world.geojson but not in zones.json"
import unittest
from electricitymap.contrib.config import EXCHANGES_CONFIG, ZONES_CONFIG
ZONE_KEYS = ZONES_CONFIG.keys()
class ExchangeJsonTestcase(unittest.TestCase):
def test_all_zones_in_zones_json(self):
for zone_key, values in EXCHANGES_CONFIG.items():
self.assertIn("->", zone_key)
for zone in zone_key.split("->"):
if zone == "US":
# Old US zone that we ignore
continue
self.assertIn(zone, ZONE_KEYS)
if __name__ == "__main__":
unittest.main(buffer=True)