def test_eat_and_move_east(self):
blobble_env = BlobbleEnv()
blobble_env.reset()
observation = blobble_env.reset_test([3, 3, 5], TEST_FOOD)
print(observation)
self.assertEqual(1, observation[3], 'Should be food of value 1')
observation, reward, done, _ = blobble_env.step(7) # Eat food, Go East
print(observation)
observation, reward, done, _ = blobble_env.step(4) # Go West back to original (no food) spot
print(observation)
self.assertEqual(0, observation[3], 'Should be no food')
observation, reward, done, _ = blobble_env.step(4)
blobble_env.render(mode='human')
def create_blobble_video(video_filename, num_episodes=1, fps=30):
filename = video_filename + ".mp4"
blobble_env = BlobbleEnv()
print('Observation space: '+str(blobble_env.observation_space))
print('Action space: '+str(blobble_env.action_space))
with imageio.get_writer(filename, fps=fps) as video:
for i in range(num_episodes-1):
print(i)
blobble_env.reset()
done = False
while not done:
action = np.random.randint(8)
print('Next Action: ', action)
observation, reward, done, _ = blobble_env.step(action)
blobble_env.render_print()
video.append_data(blobble_env.render(mode='rgb_array'))
blobble_env.close()
def test_render_human(self):
blobble_env = BlobbleEnv()
blobble_env.reset()
blobble_env.reset_test([-2, 5, 5], TEST_FOOD)
blobble_env.render(mode='human')