Browse Source

fix pylint

tags/v0.3.0-alpha
luopengting 5 years ago
parent
commit
ee07d8364d
1 changed files with 6 additions and 6 deletions
  1. +6
    -6
      tests/ut/datavisual/data_transform/test_events_data.py

+ 6
- 6
tests/ut/datavisual/data_transform/test_events_data.py View File

@@ -151,9 +151,9 @@ class TestEventsData:
# Current steps should be: [1, 2, 3, 4, 5, 10]
steps = [1, 2, 3, 4, 5, 10]
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]
for step in new_steps_2:
@@ -165,6 +165,6 @@ class TestEventsData:
steps = [1, 2, 3, 5, 7, 10, 11]
new_steps_2.extend(new_steps_1)
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)

Loading…
Cancel
Save