def test_single_cut_sampler_len(): # total duration is 55 seconds # each second has 100 frames cuts = CutSet.from_cuts( dummy_cut(idx, duration=float(idx)) for idx in range(1, 11)) sampler = SingleCutSampler(cuts, shuffle=True, max_frames=10 * 100, max_cuts=6) for epoch in range(5): assert len(sampler) == len([batch for batch in sampler]) sampler.set_epoch(epoch)
def test_single_cut_sampler_order_is_deterministic_given_epoch(): cut_set = DummyManifest(CutSet, begin_id=0, end_id=100) sampler = SingleCutSampler( cut_set, shuffle=True, # Set an effective batch size of 10 cuts, as all have 1s duration == 100 frames # This way we're testing that it works okay when returning multiple batches in # a full epoch. max_frames=1000) sampler.set_epoch(42) # calling the sampler twice without epoch update gives identical ordering assert [item for item in sampler] == [item for item in sampler]
def test_single_cut_sampler_order_differs_between_epochs(): cut_set = DummyManifest(CutSet, begin_id=0, end_id=100) sampler = SingleCutSampler( cut_set, shuffle=True, # Set an effective batch size of 10 cuts, as all have 1s duration == 100 frames # This way we're testing that it works okay when returning multiple batches in # a full epoch. max_frames=1000) last_order = [item for item in sampler] for epoch in range(1, 6): sampler.set_epoch(epoch) new_order = [item for item in sampler] assert new_order != last_order last_order = new_order