Skip to content

Commit 0c34a57

Browse files
committed
Fixed worker name key error for unutilized worker in acceptance test TestIsoScope.test_single_scope_subset_of_workers_utilized.
1 parent 776c301 commit 0c34a57

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

testing/acceptance_test.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1306,9 +1306,13 @@ def test(i):
13061306
counts_by_worker = get_workers_and_test_count_by_prefix(
13071307
"test_a.py::test", result.outlines
13081308
)
1309-
assert counts_by_worker["gw0"] in (0, num_tests)
1310-
assert counts_by_worker["gw1"] in (0, num_tests)
1311-
assert counts_by_worker["gw0"] + counts_by_worker["gw1"] == num_tests
1309+
assert counts_by_worker.setdefault("gw0", None) in (None, num_tests)
1310+
assert counts_by_worker.setdefault("gw1", None) in (None, num_tests)
1311+
assert counts_by_worker.setdefault(
1312+
"gw0", 0
1313+
) + counts_by_worker.setdefault(
1314+
"gw1", 0
1315+
) == num_tests
13121316

13131317

13141318
class TestLoadScope:

0 commit comments

Comments
 (0)