def test_stop_agents_raise_ValueError(self, fixture_stop_agents,
fixture_raise_errors):
"""
Raises an exception when the input DataFrame columns `vx` or `vy`
does not exist.
"""
df = DataFrame(pytest.data_wrong_name_columns)
with pytest.raises(ValueError):
AgentMovement.stop_agents(df, pytest.list_index_stopped)
def test_stop_agents(self, fixture_stop_agents):
"""Stops agent movement with a correct list of index."""
df = DataFrame(pytest.data)
df = AgentMovement.stop_agents(df, pytest.list_index_stopped)
for i in range(len(pytest.list_index_stopped)):
assert df['vx'][i] == df['vy'][i] == 0