Skip to content

Commit

Permalink
s/#/value
Browse files Browse the repository at this point in the history
  • Loading branch information
mag1cp1n authored Oct 20, 2023
1 parent dd0c14e commit 933f054
Showing 1 changed file with 10 additions and 10 deletions.
20 changes: 10 additions & 10 deletions tests/unit/legate/driver/test_command.py
Original file line number Diff line number Diff line change
Expand Up @@ -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", "2", "-ll:bgworkpin", "1")
assert result == ("-ll:bgwork", value, "-ll:bgworkpin", "1")
else:
assert result == ("-ll:bgwork", "2")
assert result == ("-ll:bgwork", value)

@pytest.mark.parametrize("rank_var", RANK_ENV_VARS)
@pytest.mark.parametrize("rank", ("0", "1", "2"))
Expand Down Expand Up @@ -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", "2", "-ll:bgworkpin", "1")
assert result == ("-ll:bgwork", value, "-ll:bgworkpin", "1")
else:
assert result == ("-ll:bgwork", "2")
assert result == ("-ll:bgwork", value)

@pytest.mark.parametrize("rank_var", RANK_ENV_VARS)
@pytest.mark.parametrize("rank", ("0", "1", "2"))
Expand Down Expand Up @@ -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", "2", "-ll:bgworkpin", "1")
assert result == ("-ll:bgwork", value, "-ll:bgworkpin", "1")
else:
assert result == ("-ll:bgwork", "2")
assert result == ("-ll:bgwork", value)

@pytest.mark.parametrize("launch", ("mpirun", "jsrun", "srun"))
def test_utility_1_multi_rank_with_launcher_and_ucx(
Expand Down Expand Up @@ -1193,9 +1193,9 @@ def test_utility_n_multi_rank_no_launcher(
result = m.cmd_bgwork(config, system, launcher)

if "ucx" in install_info.networks:
assert result == ("-ll:bgwork", "2", "-ll:bgworkpin", "1")
assert result == ("-ll:bgwork", value, "-ll:bgworkpin", "1")
else:
assert result == ("-ll:bgwork", "2")
assert result == ("-ll:bgwork", value)

@pytest.mark.parametrize("rank_var", RANK_ENV_VARS)
@pytest.mark.parametrize("rank", ("0", "1", "2"))
Expand All @@ -1213,9 +1213,9 @@ def test_utility_n_multi_rank_no_launcher_and_ucx(
install_info.networks[:] = networks_orig[:]

if "ucx" in install_info.networks:
assert result == ("-ll:bgwork", "2", "-ll:bgworkpin", "1")
assert result == ("-ll:bgwork", value, "-ll:bgworkpin", "1")
else:
assert result == ("-ll:bgwork", "2")
assert result == ("-ll:bgwork", value)

@pytest.mark.parametrize("launch", ("mpirun", "jsrun", "srun"))
@pytest.mark.parametrize("value", ("2", "3", "10"))
Expand Down

0 comments on commit 933f054

Please sign in to comment.