Fix flaky test

Kobzol · Kobzol · commit 421e1b25d934 · 2025-07-24T17:48:33.000+02:00
diff --git a/tests/test_coupling.py b/tests/test_coupling.py
@@ -62,7 +62,7 @@ def test_coupling_alloc1(hq_env: HqEnv):
         args=["--coupling=cpus,foo", "--resource=ddd=sum(123)", "--resource=foo=[[10,20,30,40],[50,60,70,80]]"],
     )
 
-    wait_for_job_state(hq_env, 1, "FINISHED")
+    wait_for_job_state(hq_env, [1, 2], "FINISHED")
 
     def get_groups(data):
         cpus, foos = (x.split(",") for x in data.rstrip().split(";"))

Original file line number	Diff line number	Diff line change
`@@ -62,7 +62,7 @@ def test_coupling_alloc1(hq_env: HqEnv):`
`62`	`62`	`args=["--coupling=cpus,foo", "--resource=ddd=sum(123)", "--resource=foo=[[10,20,30,40],[50,60,70,80]]"],`
`63`	`63`	`)`
`64`	`64`
`65`		`- wait_for_job_state(hq_env, 1, "FINISHED")`
	`65`	`+ wait_for_job_state(hq_env, [1, 2], "FINISHED")`
`66`	`66`
`67`	`67`	`def get_groups(data):`
`68`	`68`	`cpus, foos = (x.split(",") for x in data.rstrip().split(";"))`