def test_shuffle(manifest_type):
data = DummyManifest(manifest_type, begin_id=0, end_id=4)
for idx, item in enumerate(data):
item.duration = idx
expected_durations = [2, 1, 3, 0]
rng = random.Random(42)
eager_result = data.shuffle(rng=rng)
assert [item.duration for item in eager_result] == list(expected_durations)
with as_lazy(data) as lazy_data:
lazy_result = lazy_data.shuffle(rng=rng)
assert [item.duration
for item in lazy_result] == list(expected_durations)
def test_shuffle(manifest_type, rng):
any_set = DummyManifest(manifest_type, begin_id=0, end_id=200)
shuffled = any_set.shuffle(rng=rng)
assert list(any_set.ids) != list(shuffled.ids)
assert set(any_set.ids) == set(shuffled.ids)
