| @@ -151,9 +151,9 @@ class TestEventsData: | |||||
| # Current steps should be: [1, 2, 3, 4, 5, 10] | # Current steps should be: [1, 2, 3, 4, 5, 10] | ||||
| steps = [1, 2, 3, 4, 5, 10] | steps = [1, 2, 3, 4, 5, 10] | ||||
| samples = ev_data._reservoir_by_tag[tag].samples() | samples = ev_data._reservoir_by_tag[tag].samples() | ||||
| for i in range(len(samples)): | |||||
| filename = file1 if samples[i].step < 5 else file2 | |||||
| assert samples[i] == _Tensor(wall_time, steps[i], value, filename) | |||||
| for step, sample in zip(steps, samples): | |||||
| filename = file1 if sample.step < 5 else file2 | |||||
| assert sample == _Tensor(wall_time, step, value, filename) | |||||
| new_steps_2 = [7, 11, 3] | new_steps_2 = [7, 11, 3] | ||||
| for step in new_steps_2: | for step in new_steps_2: | ||||
| @@ -165,6 +165,6 @@ class TestEventsData: | |||||
| steps = [1, 2, 3, 5, 7, 10, 11] | steps = [1, 2, 3, 5, 7, 10, 11] | ||||
| new_steps_2.extend(new_steps_1) | new_steps_2.extend(new_steps_1) | ||||
| samples = ev_data._reservoir_by_tag[tag].samples() | samples = ev_data._reservoir_by_tag[tag].samples() | ||||
| for i in range(len(samples)): | |||||
| filename = file2 if samples[i].step in new_steps_2 else file1 | |||||
| assert samples[i] == _Tensor(wall_time, steps[i], value, filename) | |||||
| for step, sample in zip(steps, samples): | |||||
| filename = file2 if sample.step in new_steps_2 else file1 | |||||
| assert sample == _Tensor(wall_time, step, value, filename) | |||||