diff --git a/tests/unit/legate/driver/test_command.py b/tests/unit/legate/driver/test_command.py index be74c755d..94b779ea8 100644 --- a/tests/unit/legate/driver/test_command.py +++ b/tests/unit/legate/driver/test_command.py @@ -1098,9 +1098,9 @@ def test_default_multi_rank( result = m.cmd_bgwork(config, system, launcher) if "ucx" in install_info.networks: - assert result == ("-ll:bgwork", value, "-ll:bgworkpin", "1") + assert result == ("-ll:bgwork", "2", "-ll:bgworkpin", "1") else: - assert result == ("-ll:bgwork", value) + assert result == ("-ll:bgwork", "2") @pytest.mark.parametrize("rank_var", RANK_ENV_VARS) @pytest.mark.parametrize("rank", ("0", "1", "2")) @@ -1130,9 +1130,9 @@ def test_utility_1_multi_rank_no_launcher( result = m.cmd_bgwork(config, system, launcher) if "ucx" in install_info.networks: - assert result == ("-ll:bgwork", value, "-ll:bgworkpin", "1") + assert result == ("-ll:bgwork", "2"", "-ll:bgworkpin", "1") else: - assert result == ("-ll:bgwork", value) + assert result == ("-ll:bgwork", "2"") @pytest.mark.parametrize("rank_var", RANK_ENV_VARS) @pytest.mark.parametrize("rank", ("0", "1", "2")) @@ -1161,9 +1161,9 @@ def test_utility_1_multi_rank_with_launcher( result = m.cmd_bgwork(config, system, launcher) if "ucx" in install_info.networks: - assert result == ("-ll:bgwork", value, "-ll:bgworkpin", "1") + assert result == ("-ll:bgwork", "2"", "-ll:bgworkpin", "1") else: - assert result == ("-ll:bgwork", value) + assert result == ("-ll:bgwork", "2"") @pytest.mark.parametrize("launch", ("mpirun", "jsrun", "srun")) def test_utility_1_multi_rank_with_launcher_and_ucx(