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}