17 lines
496 B
Python
17 lines
496 B
Python
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
|