23 lines
851 B
Python
23 lines
851 B
Python
from energysphere.models import LocalProblem
|
|
from energysphere.energi_bridge import EnergiBridgeMapper
|
|
|
|
|
|
def test_map_local_problem_to_object():
|
|
lp = LocalProblem(site_id="site-42", objective="min-cost", parameters={"load": 2.5})
|
|
mapper = EnergiBridgeMapper(namespace="test-ns")
|
|
obj = mapper.map_local_problem(lp)
|
|
assert obj["type"] == "Object"
|
|
assert obj["site_id"] == "site-42"
|
|
assert obj["namespace"] == "test-ns"
|
|
assert obj["version"] == 1
|
|
|
|
|
|
def test_map_object_to_local_problem_roundtrip():
|
|
lp = LocalProblem(site_id="site-7", objective="min-cost", parameters={"p": 1})
|
|
mapper = EnergiBridgeMapper()
|
|
obj = mapper.map_local_problem(lp)
|
|
lp2 = mapper.map_object_to_local_problem(obj)
|
|
assert lp2.site_id == lp.site_id
|
|
assert lp2.objective == lp.objective
|
|
assert lp2.parameters == lp.parameters
|