citygrid-policy-driven-fede.../tests/test_adapter_interop.py

15 lines
852 B
Python

from citygrid.adapters.der_controller import DerControllerAdapter
from citygrid.adapters.water_pump_controller import WaterPumpControllerAdapter
from citygrid.bridge.energi_bridge import EnergiBridge
def test_adapters_to_canonical_and_ack():
der = DerControllerAdapter()
water = WaterPumpControllerAdapter()
lp_der = der.build_local_problem()
lp_water = water.build_local_problem()
canon_der = EnergiBridge.to_canonical({"id": lp_der.id, "domain": lp_der.domain, "assets": lp_der.assets, "objective": lp_der.objective, "constraints": lp_der.constraints})
canon_water = EnergiBridge.to_canonical({"id": lp_water.id, "domain": lp_water.domain, "assets": lp_water.assets, "objective": lp_water.objective, "constraints": lp_water.constraints})
assert canon_der.domain == "electricity"
assert canon_water.domain == "water"