Skip to content

Commit 8649f6f

Browse files
committed
work around timing issue in whisper reader interval tests
1 parent 4c5a7c4 commit 8649f6f

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

webapp/tests/test_readers_whisper.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -79,14 +79,14 @@ def test_GzippedWhisperReader_get_intervals(self):
7979
ts = int(time.time())
8080
intervals = reader.get_intervals()
8181
for interval in intervals:
82-
self.assertEqual(int(interval.start), ts-60)
83-
self.assertEqual(int(interval.end), ts)
82+
self.assertEqual(int(interval.start), ts - 60)
83+
self.assertIn(int(interval.end), [ts, ts - 1])
8484

8585
# read it again to validate cache works
8686
intervals = reader.get_intervals()
8787
for interval in intervals:
88-
self.assertEqual(int(interval.start),ts-60)
89-
self.assertEqual(int(interval.end), ts)
88+
self.assertEqual(int(interval.start),ts - 60)
89+
self.assertIn(int(interval.end), [ts, ts - 1])
9090

9191
# Confirm fetch works.
9292
def test_GzippedWhisperReader_fetch(self):
@@ -120,14 +120,14 @@ def test_WhisperReader_get_intervals(self):
120120
ts = int(time.time())
121121
intervals = reader.get_intervals()
122122
for interval in intervals:
123-
self.assertEqual(int(interval.start),ts-60)
124-
self.assertEqual(int(interval.end), ts)
123+
self.assertEqual(int(interval.start),ts - 60)
124+
self.assertIn(int(interval.end), [ts, ts - 1])
125125

126126
# read it again to validate cache works
127127
intervals = reader.get_intervals()
128128
for interval in intervals:
129-
self.assertEqual(int(interval.start),ts-60)
130-
self.assertEqual(int(interval.end), ts)
129+
self.assertEqual(int(interval.start),ts - 60)
130+
self.assertIn(int(interval.end), [ts, ts - 1])
131131

132132
# Confirm get_raw_step works
133133
def test_WhisperReader_get_raw_step(self):

0 commit comments

Comments
 (0)