Skip to content

Commit

Permalink
Replace constants where value is not defined.
Browse files Browse the repository at this point in the history
  • Loading branch information
mag1cp1n authored Oct 23, 2023
1 parent ac12c5d commit a01f71c
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 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", 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"))
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", 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"))
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", 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(
Expand Down

0 comments on commit a01f71c

Please sign in to comment.