Skip to content

Commit

Permalink
Fix test level for switch default env (#906)
Browse files Browse the repository at this point in the history
  • Loading branch information
carolineechen committed Jun 18, 2024
1 parent e95e4f9 commit 302b111
Showing 1 changed file with 16 additions and 16 deletions.
32 changes: 16 additions & 16 deletions tests/test_resources/test_clusters/test_cluster.py
Original file line number Diff line number Diff line change
Expand Up @@ -794,6 +794,22 @@ def test_default_conda_env_created(self, cluster):
assert isinstance(cluster.get(cluster.default_env.name), rh.CondaEnv)

@pytest.mark.level("local")
def test_default_env_var_run(self, cluster):
env_vars = cluster.default_env.env_vars
if not env_vars:
pytest.skip("No env vars in default env")

assert env_vars
for var in env_vars.keys():
res = cluster.run([f"echo ${var}"], env=cluster.default_env)
assert res[0][0] == 0
assert env_vars[var] in res[0][1]

get_env_var_cpu = rh.function(_get_env_var_value).to(system=cluster)
for var in env_vars.keys():
assert get_env_var_cpu(var) == env_vars[var]

@pytest.mark.level("release")
def test_switch_default_env(self, cluster):
# test setting a new default env, w/o restarting the runhouse server
test_env = cluster.default_env
Expand All @@ -815,19 +831,3 @@ def test_switch_default_env(self, cluster):
# set it back
cluster.default_env = test_env
cluster.delete(new_env.name)

@pytest.mark.level("local")
def test_default_env_var_run(self, cluster):
env_vars = cluster.default_env.env_vars
if not env_vars:
pytest.skip("No env vars in default env")

assert env_vars
for var in env_vars.keys():
res = cluster.run([f"echo ${var}"], env=cluster.default_env)
assert res[0][0] == 0
assert env_vars[var] in res[0][1]

get_env_var_cpu = rh.function(_get_env_var_value).to(system=cluster)
for var in env_vars.keys():
assert get_env_var_cpu(var) == env_vars[var]

0 comments on commit 302b111

Please sign in to comment.