catopt-graph-graph-calculus.../tests/test_contract_registry.py

17 lines
562 B
Python

import pytest
from core.contracts import ContractRegistry
def test_contract_registry_basic():
reg = ContractRegistry()
reg.add_contract("LocalProblem", "v1", {"fields": ["asset_id", "payload"]})
reg.add_contract("SharedVariables", "v1", {"fields": ["iter_id", "values"]})
c1 = reg.get_contract("LocalProblem", "v1")
c2 = reg.get_contract("SharedVariables", "v1")
assert c1 is not None
assert c2 is not None
assert c1.name == "LocalProblem" and c1.version == "v1"
assert c2.name == "SharedVariables" and c2.version == "v1"