Skip to content

Commit

Permalink
Reinstate missing assert
Browse files Browse the repository at this point in the history
  • Loading branch information
oyvindeide committed Jan 26, 2024
1 parent a8ca0c5 commit 05c9f12
Showing 1 changed file with 8 additions and 2 deletions.
10 changes: 8 additions & 2 deletions tests/integration_tests/status/test_tracking_integration.py
Original file line number Diff line number Diff line change
Expand Up @@ -228,20 +228,26 @@ def test_tracking(
snapshots = {}

thread.join()

last_progress = 0
while not queue.empty():
event = queue.get()
if isinstance(event, FullSnapshotEvent):
last_progress = (
event.progress if event.progress > last_progress else last_progress
)
snapshots[event.iteration] = event.snapshot
if (
isinstance(event, SnapshotUpdateEvent)
and event.partial_snapshot is not None
):
snapshots[event.iteration].merge(event.partial_snapshot.data())
last_progress = (
event.progress if event.progress > last_progress else last_progress
)
if isinstance(event, EndEvent):
pass

# assert tracker._progress() == progress
assert last_progress == progress

assert len(snapshots) == num_iters
for snapshot in snapshots.values():
Expand Down

0 comments on commit 05c9f12

Please sign in to comment.