def test_load_scenario():
s = """
inputs:
origin: hand-made
events:
- id: w1
delay: 30
- id: e1
actions:
- type: remove_agent
agent: a005
- type: remove_agent
agent: a068
- id: w2
delay: 10
- id: e2
actions:
- type: remove_agent
agent: a026
- type: remove_agent
agent: a056
"""
scenario = load_scenario(s)
assert len(scenario.events) == 4
assert scenario.events[0].is_delay
assert len(scenario.events[1].actions) == 2
assert scenario.events[3].actions[1].type == "remove_agent"
assert scenario.events[3].actions[0].args["agent"] == "a026"
def test_yaml_scenario_one_event():
events = [
DcopEvent("1", actions=[EventAction("remove_agent", agent="a01")])
]
scenario = Scenario(events)
scenario_str = yaml_scenario(scenario)
obtained = load_scenario(scenario_str)
assert len(obtained.events) == 1
assert len(obtained.events[0].actions) == 1
def test_yaml_scenario_one_event_two_actions():
events = [
DcopEvent(
"1",
actions=[
EventAction("remove_agent", agent="a01"),
EventAction("remove_agent", agent="a05"),
],
)
]
scenario = Scenario(events)
scenario_str = yaml_scenario(scenario)
obtained = load_scenario(scenario_str)
assert len(obtained.events) == 1
assert len(obtained.events[0].actions) == 2
assert obtained.events[0].actions[1].type == "remove_agent"
assert obtained.events[0].actions[1].args["agent"] == "a05"
def test_yaml_scenario_two_events():
events = [
DcopEvent(
"1",
actions=[
EventAction("remove_agent", agent="a01"),
EventAction("remove_agent", agent="a05"),
],
),
DcopEvent("2", delay=30),
]
scenario = Scenario(events)
scenario_str = yaml_scenario(scenario)
obtained = load_scenario(scenario_str)
assert len(obtained.events) == 2
assert obtained.events[1].is_delay
assert not obtained.events[0].is_delay
