test fix

quantumlib · Aug 7, 2024 · 88c1e4e · 88c1e4e
1 parent 4269062
commit 88c1e4e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/glue/sample/src/sinter/_collection/_collection_manager_test.py b/glue/sample/src/sinter/_collection/_collection_manager_test.py
@@ -80,7 +80,7 @@ def test_manager():
     _assert_drain_queue(manager.worker_states[0].input_queue, [
         (
             'change_job',
-            (t0, sinter.CollectionOptions(max_errors=100)),
+            (t0, sinter.CollectionOptions(max_errors=100), 100),
         ),
         (
             'accept_shots',
@@ -91,7 +91,7 @@ def test_manager():
         ('compute_strong_id', t0),
         (
             'change_job',
-            (t1, sinter.CollectionOptions(max_errors=10000000)),
+            (t1, sinter.CollectionOptions(max_errors=10000000), 10000000),
         ),
         (
             'accept_shots',
@@ -102,7 +102,7 @@ def test_manager():
         ('compute_strong_id', t1),
         (
             'change_job',
-            (t1, sinter.CollectionOptions(max_errors=10000000)),
+            (t1, sinter.CollectionOptions(max_errors=10000000), 10000000),
         ),
         (
             'accept_shots',
@@ -279,9 +279,9 @@ def test_manager():
         ('return_shots', (t0.strong_id(), 66666666)),
     ])
     _assert_drain_queue(manager.worker_states[1].input_queue, [
-        ('change_job', (t0, sinter.CollectionOptions(max_errors=100))),
+        ('change_job', (t0, sinter.CollectionOptions(max_errors=100), 100)),
     ])
     _assert_drain_queue(manager.worker_states[2].input_queue, [
         ('return_shots', (t1.strong_id(), 1000000)),
-        ('change_job', (t0, sinter.CollectionOptions(max_errors=100))),
+        ('change_job', (t0, sinter.CollectionOptions(max_errors=100), 100)),
     ])
diff --git a/glue/sample/src/sinter/_collection/_collection_worker_test.py b/glue/sample/src/sinter/_collection/_collection_worker_test.py
@@ -69,7 +69,7 @@ def test_worker_stop():
     )
     handler.expected_task = t0
 
-    _put_wait_not_empty(inp, ('change_job', (t0, sinter.CollectionOptions(max_errors=100_000_000))))
+    _put_wait_not_empty(inp, ('change_job', (t0, sinter.CollectionOptions(max_errors=100_000_000), 100_000_000)))
     assert worker.process_messages() == 1
     _assert_drain_queue(out, [('changed_job', 5, (t0.strong_id(),))])
 
@@ -105,7 +105,7 @@ def test_worker_skip_work():
         json_metadata={'a': 3},
     )
     handler.expected_task = t0
-    _put_wait_not_empty(inp, ('change_job', (t0, sinter.CollectionOptions(max_errors=100_000_000))))
+    _put_wait_not_empty(inp, ('change_job', (t0, sinter.CollectionOptions(max_errors=100_000_000), 100_000_000)))
     assert worker.process_messages() == 1
     _assert_drain_queue(out, [('changed_job', 5, (t0.strong_id(),))])
 
@@ -161,7 +161,7 @@ def test_worker_finish_work():
         json_metadata={'a': 3},
     )
     handler.expected_task = ta
-    _put_wait_not_empty(inp, ('change_job', (ta, sinter.CollectionOptions(max_errors=100_000_000))))
+    _put_wait_not_empty(inp, ('change_job', (ta, sinter.CollectionOptions(max_errors=100_000_000), 100_000_000)))
     _put_wait_not_empty(inp, ('accept_shots', (ta.strong_id(), 10000)))
     assert worker.process_messages() == 2
     _assert_drain_queue(out, [