test_utility_n_multi_rank_with_launcher: Add logic

nv-legate · Oct 23, 2023 · ac12c5d · ac12c5d
1 parent 35c7055
commit ac12c5d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/unit/legate/driver/test_command.py b/tests/unit/legate/driver/test_command.py
@@ -1228,7 +1228,10 @@ def test_utility_n_multi_rank_with_launcher(
 
         result = m.cmd_bgwork(config, system, launcher)
 
-        assert result == ("-ll:bgwork", value)
+        if "ucx" in install_info.networks:
+            assert result == ("-ll:bgwork", value, "-ll:bgworkpin", "1")
+        else:
+            assert result == ("-ll:bgwork", value)
 
     @pytest.mark.parametrize("launch", ("mpirun", "jsrun", "srun"))
     @pytest.mark.parametrize("value", ("2", "3", "10"))