from idea165_commonsgrid_community_managed.models import LocalProblem
from idea165_commonsgrid_community_managed.adapters import DERAdapter, BatteryAdapter


def test_adapters_to_shared():
    lp = LocalProblem(neighborhood_id="nb1", demand_kw=10.0, pv_kw=6.0, storage_kwh=5.0, evs=2)
    der = DERAdapter()
    bat = BatteryAdapter()
    assert der.to_shared(lp)["type"] == "DER"
    assert bat.to_shared(lp)["type"] == "Battery"