12 lines
433 B
Python
12 lines
433 B
Python
from energysphere.registry import GraphContractRegistry
|
|
from energysphere.models import LocalProblem
|
|
|
|
|
|
def test_registry_register_and_list():
|
|
reg = GraphContractRegistry()
|
|
lp = LocalProblem(site_id="site-1", objective="min-cost", parameters={"load": 1.0})
|
|
reg.register_local_problem("contract-site-1", lp)
|
|
contracts = reg.list_contracts()
|
|
assert len(contracts) == 1
|
|
assert contracts[0]["site_id"] == "site-1"
|