def test_side_effects():
class DoneState(State):
def setup(self, builder):
super().setup(builder)
self.population_view = builder.population.get_view(['count'])
def _transition_side_effect(self, index, event_time):
pop = self.population_view.get(index)
self.population_view.update(pop['count'] + 1)
done_state = DoneState('done')
start_state = State('start')
start_state.add_transition(done_state)
done_state.add_transition(start_state)
machine = Machine('state', states=[start_state, done_state])
simulation = setup_simulation([machine, _population_fixture('state', 'start'), _population_fixture('count', 0)])
event_time = simulation.clock.time + simulation.clock.step_size
machine.transition(simulation.population.population.index, event_time)
assert np.all(simulation.population.population['count'] == 1)
machine.transition(simulation.population.population.index, event_time)
assert np.all(simulation.population.population['count'] == 1)
machine.transition(simulation.population.population.index, event_time)
assert np.all(simulation.population.population['count'] == 2)
def test_no_null_transition(base_config):
base_config.update({'population': {'population_size': 10000}})
a_state = State('a')
b_state = State('b')
start_state = State('start')
a_transition = Transition(
start_state,
a_state,
probability_func=lambda index: pd.Series(0.5, index=index))
b_transition = Transition(
start_state,
b_state,
probability_func=lambda index: pd.Series(0.5, index=index))
start_state.transition_set.allow_null_transition = False
start_state.transition_set.extend((a_transition, b_transition))
machine = Machine('state')
machine.states.extend([start_state, a_state, b_state])
simulation = InteractiveContext(
components=[machine, _population_fixture('state', 'start')],
configuration=base_config)
event_time = simulation._clock.time + simulation._clock.step_size
machine.transition(simulation.get_population().index, event_time)
a_count = (simulation.get_population().state == 'a').sum()
assert round(a_count / len(simulation.get_population()), 1) == 0.5
def test_transition():
done_state = State('done')
start_state = State('start')
start_state.add_transition(done_state)
machine = Machine('state', states=[start_state, done_state])
simulation = setup_simulation([machine, _population_fixture('state', 'start')])
event_time = simulation.clock.time + simulation.clock.step_size
machine.transition(simulation.population.population.index, event_time)
assert np.all(simulation.population.population.state == 'done')
def test_transition():
done_state = State('done')
start_state = State('start')
start_state.add_transition(done_state)
machine = Machine('state', states=[start_state, done_state])
simulation = InteractiveContext(
components=[machine, _population_fixture('state', 'start')])
event_time = simulation._clock.time + simulation._clock.step_size
machine.transition(simulation.get_population().index, event_time)
assert np.all(simulation.get_population().state == 'done')
def test_transition():
done_state = State('done')
start_state = State('start')
done_transition = Transition(done_state,
lambda agents: np.full(len(agents), 1.0))
start_state.transition_set.append(done_transition)
machine = Machine('state')
machine.states.extend([start_state, done_state])
simulation = setup_simulation(
[machine, _population_fixture('state', 'start')])
machine.transition(simulation.population.population.index)
assert np.all(simulation.population.population.state == 'done')
def test_side_effects():
class DoneState(State):
@uses_columns(['count'])
def _transition_side_effect(self, index, population_view):
pop = population_view.get(index)
population_view.update(pop['count'] + 1)
done_state = DoneState('done')
start_state = State('start')
done_transition = Transition(done_state,
lambda agents: np.full(len(agents), 1.0))
start_state.transition_set.append(done_transition)
done_state.transition_set.append(done_transition)
machine = Machine('state')
machine.states.extend([start_state, done_state])
simulation = setup_simulation([
machine,
_population_fixture('state', 'start'),
_population_fixture('count', 0)
])
machine.transition(simulation.population.population.index)
assert np.all(simulation.population.population['count'] == 1)
machine.transition(simulation.population.population.index)
assert np.all(simulation.population.population['count'] == 2)
def test_no_null_transition(base_config):
base_config.population.update({'population_size': 10000})
a_state = State('a')
b_state = State('b')
start_state = State('start')
a_transition = Transition(start_state, a_state)
b_transition = Transition(start_state, b_state)
start_state.transition_set.allow_null_transition = False
start_state.transition_set.extend((a_transition, b_transition))
machine = Machine('state')
machine.states.extend([start_state, a_state, b_state])
simulation = setup_simulation([machine, _population_fixture('state', 'start')], base_config)
event_time = simulation.clock.time + simulation.clock.step_size
machine.transition(simulation.population.population.index, event_time)
a_count = (simulation.population.population.state == 'a').sum()
assert round(a_count/len(simulation.population.population), 1) == 0.5
def test_no_null_transition():
a_state = State('a')
b_state = State('b')
start_state = State('start')
a_transition = Transition(a_state)
b_transition = Transition(b_state)
start_state.transition_set.allow_null_transition = False
start_state.transition_set.extend((a_transition, b_transition))
machine = Machine('state')
machine.states.extend([start_state, a_state, b_state])
simulation = setup_simulation(
[machine, _population_fixture('state', 'start')],
population_size=10000)
machine.transition(simulation.population.population.index)
a_count = (simulation.population.population.state == 'a').sum()
assert round(a_count / len(simulation.population.population), 1) == 0.5
def test_choice():
a_state = State('a')
b_state = State('b')
start_state = State('start')
a_transition = Transition(a_state,
lambda agents: np.full(len(agents), 0.5))
b_transition = Transition(b_state,
lambda agents: np.full(len(agents), 0.5))
start_state.transition_set.extend((a_transition, b_transition))
machine = Machine('state')
machine.states.extend([start_state, a_state, b_state])
simulation = setup_simulation(
[machine, _population_fixture('state', 'start')],
population_size=10000)
machine.transition(simulation.population.population.index)
a_count = (simulation.population.population.state == 'a').sum()
assert round(a_count / len(simulation.population.population), 1) == 0.5
def test_null_transition(base_config):
base_config.population.update({'population_size': 10000})
a_state = State('a')
start_state = State('start')
start_state.add_transition(a_state, probability_func=lambda agents: np.full(len(agents), 0.5))
start_state.allow_self_transitions()
machine = Machine('state', states=[start_state, a_state])
simulation = setup_simulation([machine, _population_fixture('state', 'start')], base_config)
event_time = simulation.clock.time + simulation.clock.step_size
machine.transition(simulation.population.population.index, event_time)
a_count = (simulation.population.population.state == 'a').sum()
assert round(a_count/len(simulation.population.population), 1) == 0.5
def test_choice(base_config):
base_config.update({'population': {'population_size': 10000}})
a_state = State('a')
b_state = State('b')
start_state = State('start')
start_state.add_transition(
a_state, probability_func=lambda agents: np.full(len(agents), 0.5))
start_state.add_transition(
b_state, probability_func=lambda agents: np.full(len(agents), 0.5))
machine = Machine('state', states=[start_state, a_state, b_state])
simulation = InteractiveContext(
components=[machine, _population_fixture('state', 'start')],
configuration=base_config)
event_time = simulation._clock.time + simulation._clock.step_size
machine.transition(simulation.get_population().index, event_time)
a_count = (simulation.get_population().state == 'a').sum()
assert round(a_count / len(simulation.get_population()), 1) == 0.5
def test_null_transition():
a_state = State('a')
start_state = State('start')
start_state.add_transition(
a_state, probability_func=lambda agents: np.full(len(agents), 0.5))
start_state.allow_self_transitions()
machine = Machine('state', states=[start_state, a_state])
simulation = setup_simulation(
[machine, _population_fixture('state', 'start')],
population_size=10000)
machine.transition(simulation.population.population.index)
a_count = (simulation.population.population.state == 'a').sum()
assert round(a_count / len(simulation.population.population), 1) == 0.5