diff --git a/tests/test_energi_bridge_roundtrip.py b/tests/test_energi_bridge_roundtrip.py new file mode 100644 index 0000000..36a4217 --- /dev/null +++ b/tests/test_energi_bridge_roundtrip.py @@ -0,0 +1,29 @@ +from gridverse.energi_bridge import EnergiBridge + + +def test_to_ir_roundtrip_basic(): + payload = {"p": 42, "q": "hello"} + ir = EnergiBridge.to_ir("LocalProblem", payload) + assert ir["ir_type"] == "LocalProblem" + assert ir["ir_payload"] == payload + + +def test_from_ir_roundtrip_basic(): + ir = { + "ir_type": "LocalProblem", + "ir_payload": {"p": 7}, + "metadata": {"source": "test"}, + } + contract_type, payload = EnergiBridge.from_ir(ir) + assert contract_type == "LocalProblem" + assert payload == {"p": 7} + + +def test_map_gridverse_to_catopt_basic(): + gridverse_contract = { + "contract_type": "LocalProblem", + "payload": {"a": 1}, + } + ir = EnergiBridge.map_gridverse_to_catopt(gridverse_contract) + assert ir["ir_type"] == "LocalProblem" + assert ir["ir_payload"] == {"a": 1}