From 88c1e4e817f4611d3ab378acd06a10df4a7cf582 Mon Sep 17 00:00:00 2001
From: Craig Gidney <craiggidney@google.com>
Date: Tue, 6 Aug 2024 20:27:19 -0700
Subject: [PATCH] test fix

---
 .../src/sinter/_collection/_collection_manager_test.py | 10 +++++-----
 .../src/sinter/_collection/_collection_worker_test.py  |  6 +++---
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/glue/sample/src/sinter/_collection/_collection_manager_test.py b/glue/sample/src/sinter/_collection/_collection_manager_test.py
index e40b83b5..c4cb359b 100644
--- 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
index 93d9fa86..db0518dc 100644
--- 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, [