Minor change to improve test clarity

2024-05-28 23:16:15 +10:00 · 2024-05-28 23:16:15 +10:00 · 5d7a4e316b
parent 621f69d98f
commit 5d7a4e316b
1 changed files with 1 additions and 0 deletions
--- a/tests/test_sampler.py
+++ b/tests/test_sampler.py
@ -85,5 +85,6 @@ def test_shuffle():
    assert len(sampler) == 6
    assert list(sampler) == [0, 1, 2, 3, 4, 5]
    sampler = EpisodeAwareSampler(episode_data_index, shuffle=True)
+    assert sampler.indices == [0, 1, 2, 3, 4, 5]
    assert len(sampler) == 6
    assert set(sampler) == {0, 1, 2, 3, 4, 5}