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