from util_simulation import sample_coords
from util_simulation import find_possible_actions
# =============================================================================
# Unit Test
# =============================================================================
# Random Initialiser
rnd.seed(42)
# Create GridWorld
grid = np.zeros((5,7))
grid[1][5] = 1
grid[0][1] = np.nan
print (grid)
g = GridWorld.create(grid, 10)
# Place blobs
g.populate()
g.show_board(figsize=(8,5))#, reward_overlay = True)
# Play multiple times
%time g.play(duration=1)
g.show_board(figsize=(8,5))#, reward_overlay = True)
# Save video
g.save_video(save_path = os.path.join(main_dir, 'Temp/temp.gif'),
start_time = 0, end_time = 20,
fps = 5, dpi = 60, figsize=(5,8))
g.save_video(save_path = os.path.join(main_dir, 'Temp/temp.mp4'),
start_time = 0, end_time = 20,
