def test_many_videos_hdf5(tmpdir): labels = Labels() filename = os.path.join(tmpdir, "test.h5") labels.videos = [Video.from_filename(f"video {i}.mp4") for i in range(3000)] Labels.save_hdf5(filename=filename, labels=labels)
def test_many_suggestions_hdf5(tmpdir): labels = Labels() filename = os.path.join(tmpdir, "test.h5") video = Video.from_filename("foo.mp4") labels.videos = [video] labels.suggestions = [SuggestionFrame(video, i) for i in range(3000)] Labels.save_hdf5(filename=filename, labels=labels)