openenergysphere-federated-.../tests/test_adapter.py

14 lines
584 B
Python

from energysphere.adapters.substation_meter import SubstationMeterAdapter
from energysphere.registry import GraphContractRegistry
from energysphere.models import LocalProblem
def test_substation_meter_adapter_generates_local_problem():
registry = GraphContractRegistry()
adapter = SubstationMeterAdapter(registry)
lp = adapter.collect_local_problem()
assert isinstance(lp, LocalProblem)
# The adapter should have registered the problem into the registry
contracts = registry.list_contracts()
assert any(c.get("site_id") == lp.site_id for c in contracts)