split actuators from energy storage
This commit is contained in:
@@ -35,7 +35,7 @@ def test_add_and_get_entity(repo):
|
||||
def test_list_entities_by_dimension(repo):
|
||||
repo.add_entity(Entity(name="A", dimension="platform"))
|
||||
repo.add_entity(Entity(name="B", dimension="platform"))
|
||||
repo.add_entity(Entity(name="C", dimension="power_source"))
|
||||
repo.add_entity(Entity(name="C", dimension="actuator"))
|
||||
|
||||
platforms = repo.list_entities(dimension="platform")
|
||||
assert len(platforms) == 2
|
||||
@@ -65,7 +65,7 @@ def test_add_and_get_domain(repo):
|
||||
|
||||
def test_combination_save_and_dedup(repo):
|
||||
e1 = repo.add_entity(Entity(name="A", dimension="platform"))
|
||||
e2 = repo.add_entity(Entity(name="B", dimension="power_source"))
|
||||
e2 = repo.add_entity(Entity(name="B", dimension="actuator"))
|
||||
|
||||
from physcom.models.combination import Combination
|
||||
combo = Combination(entities=[e1, e2])
|
||||
@@ -79,12 +79,14 @@ def test_combination_save_and_dedup(repo):
|
||||
|
||||
|
||||
def test_seed_loads(seeded_repo):
|
||||
from physcom.seed.transport_example import PLATFORMS, POWER_SOURCES, ALL_DOMAINS
|
||||
from physcom.seed.transport_example import PLATFORMS, ACTUATORS, ENERGY_STORAGES, ALL_DOMAINS
|
||||
|
||||
platforms = seeded_repo.list_entities(dimension="platform")
|
||||
power_sources = seeded_repo.list_entities(dimension="power_source")
|
||||
actuators = seeded_repo.list_entities(dimension="actuator")
|
||||
energy_storages = seeded_repo.list_entities(dimension="energy_storage")
|
||||
assert len(platforms) == len(PLATFORMS)
|
||||
assert len(power_sources) == len(POWER_SOURCES)
|
||||
assert len(actuators) == len(ACTUATORS)
|
||||
assert len(energy_storages) == len(ENERGY_STORAGES)
|
||||
|
||||
domains = seeded_repo.list_domains()
|
||||
assert len(domains) == len(ALL_DOMAINS)
|
||||
|
||||
Reference in New Issue
Block a user