fixtest

tests/trace/test_weave_client.py

@@ -1900,7 +1900,7 @@ def op_1():
     client.flush(use_progress_bar=True)
 
     # make sure there are no pending jobs
-    assert client._get_pending_jobs() == (0, 0, 0, 0)
+    assert client._get_pending_jobs()["total_jobs"] == 0
     assert client._has_pending_jobs() == False
 
 
@@ -1912,13 +1912,13 @@ def op_1():
     op_1()
 
     def fake_logger(status):
-        assert "main_jobs" in status
+        assert "job_counts" in status
 
     # flush with callback
     client.flush(callback=fake_logger)
 
     # make sure there are no pending jobs
-    assert client._get_pending_jobs() == (0, 0, 0, 0)
+    assert client._get_pending_jobs()["total_jobs"] == 0
     assert client._has_pending_jobs() == False
 
     op_1()
@@ -1927,5 +1927,5 @@ def fake_logger(status):
     client.flush(callback=fake_logger, use_progress_bar=True)
 
     # make sure there are no pending jobs
-    assert client._get_pending_jobs() == (0, 0, 0, 0)
+    assert client._get_pending_jobs()["total_jobs"] == 0
     assert client._has_pending_jobs() == False