def move_blue_to_red_2(population: Population, **kwargs):
func = lambda **kwargs: go_to(go_from="blue",
go_to="red",
**kwargs)
if population.day == 2:
return [func]
else:
return []
def move_partial(population: Population, **kwargs):
if population.day == 1:
return move_blue_to_red(population=population, **kwargs)
elif population.day == 2:
func = lambda **kwargs: go_to(go_from="red",
go_to="blue",
fraction=0.7,
**kwargs)
return [func]
else:
return []
def move_red_to_blue(**kwargs):
func = lambda **kwargs: go_to(go_from="red",
go_to="blue",
**kwargs)
return [func]