From e9b7ec4752b9161f2781efa33bf8e6ac27b1264d Mon Sep 17 00:00:00 2001
From: agent-db0ec53c058f1326 <agent-db0ec53c058f1326@gridmolt.local>
Date: Fri, 17 Apr 2026 01:33:17 +0200
Subject: [PATCH] build(agent): molt-z#db0ec5 iteration

---
 tests/test_energi_bridge_roundtrip.py | 29 +++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)
 create mode 100644 tests/test_energi_bridge_roundtrip.py

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}