def test_default_env(): assert_equal(SlurmEnvironment, get_default_environment(SlurmEnvironment)) assert_equal(InSituEnvironment, get_default_environment(InSituEnvironment)) assert_equal(BashEnvironment, get_default_environment(BashEnvironment)) with add_env(**{__CT_ENVIRONMENT_ENVVARS__: "insitu"}): assert_equal(InSituEnvironment, get_default_environment()) with add_env(**{__CT_ENVIRONMENT_ENVVARS__: "erez"}): assert_in(get_default_environment(), (SlurmEnvironment, InSituEnvironment, BashEnvironment)) expected = SlurmEnvironment if SlurmEnvironment.is_usable() \ else InSituEnvironment assert_equal(get_default_environment(), expected)
def test_monitor_incomplete_to_launchable_can_list(): monitor = Monitor(__EXP_NAME__, environment_cls=SlurmEnvironment) monitor_incomplete_to_launchable(monitor, SlurmEnvironment.is_usable())
def test_monitor_refresh_can_list(): monitor = Monitor(__EXP_NAME__, environment_cls=SlurmEnvironment) monitor_refresh(monitor, SlurmEnvironment.is_usable())
def test_list_up_jobs(): assert_is_none(InSituEnvironment.list_up_jobs()) assert_is_none(BashEnvironment.list_up_jobs()) if not SlurmEnvironment.is_usable(): assert_is_none(SlurmEnvironment.list_up_jobs())
def test_usability(): assert_equal(InSituEnvironment.is_usable(), True) assert_equal(BashEnvironment.is_usable(), True) assert_equal(SlurmEnvironment.is_usable(), shutil.which("scontrol") is not None)