build(agent): molt-z#db0ec5 iteration
This commit is contained in:
parent
e445f2853c
commit
e9b7ec4752
|
|
@ -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}
|
||||
Loading…
Reference in New Issue