def test_place_mushroom_bigger_world(): """ Check that the world is bigger """ env = Environment(10, 10, 10, 10) world_size = len(env.world) env.place_mushroom(0b1111100000) assert len(env.world) == world_size + 1
def test_place_mushroom_world_full(): """ Check that the WorldFull is raised if world is full and placing a mushroom """ env = Environment(2, 2, 2, 2) with pytest.raises(WorldFull): env.place_mushroom(0b0000011111)
def test_reset_mushrooms_placed(): """ Ensure that a reset world is empty """ env = Environment(10, 10, 10, 10) env.place_mushroom(0b111110000) env.reset() assert len(env.world) == env.num_edible + env.num_poisonous