def test_video_gif(setup):
import numpy as np
def im(x, y):
canvas = np.zeros((200, 200))
for i in range(200):
for j in range(200):
if x - 5 < i < x + 5 and y - 5 < j < y + 5:
canvas[i, j] = 1
return canvas
frames = [im(100 + i, 80) for i in range(20)]
logger.save_video(frames, "test_video.gif")
assert 'test_video.gif' in logger.glob('*.gif')
logger.save_video(frames, "/tmp/ml-logger-test/videos/test_video.gif")
assert 'ml-logger-test/videos/test_video.gif' in logger.glob(
'**/videos/*.gif', wd="/tmp")
def test_glob(setup):
kwargs = dict(query="*.pkl", wd="test-data")
print(f'globbing {kwargs["query"]} under {kwargs["wd"]}')
file_paths = logger.glob(**kwargs)
print(f"globbed file paths: {[file_paths]}")