from catopt_flow_category_theoretic_compositi import ContractRegistry, AdapterContract


def test_registry_basic():
    reg = ContractRegistry()
    ac = AdapterContract(
        platform="pytorch",
        name="gpu-adapter",
        version="0.1.0",
        capabilities=["training", "autograd"],
        contract={"interface": "local_problem"},
    )
    reg.register("pytorch/gpu-adapter", ac)
    assert reg.get("pytorch/gpu-adapter") is ac
    items = reg.list()
    assert len(items) == 1