diff --git a/tests/integration_tests/status/test_tracking_integration.py b/tests/integration_tests/status/test_tracking_integration.py index 09e0fcfabb5..ded251e1216 100644 --- a/tests/integration_tests/status/test_tracking_integration.py +++ b/tests/integration_tests/status/test_tracking_integration.py @@ -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():