diff --git a/tests/test_helpers.py b/tests/test_helpers.py index 42c56dfef32..ddf5017590c 100644 --- a/tests/test_helpers.py +++ b/tests/test_helpers.py @@ -4,6 +4,7 @@ from tests.helpers import flatten_dict from tests.helpers import isolated_environment +from tests.helpers import pbs_installer_supported_arch def test_flatten_dict() -> None: @@ -44,3 +45,15 @@ def test_isolated_environment_updates_environ() -> None: with isolated_environment(environ={"NEW_VAR": "new_value"}): assert os.environ["NEW_VAR"] == "new_value" assert "NEW_VAR" not in os.environ + +#---------------------2 new tests----------------------- + +def test_pbs_installer_supported_arch_accepts_supported_archs() -> None: + assert pbs_installer_supported_arch("x86_64") is True + assert pbs_installer_supported_arch("amd64") is True + assert pbs_installer_supported_arch("arm64") is True + + +def test_pbs_installer_supported_arch_rejects_unsupported_archs() -> None: + assert pbs_installer_supported_arch("sparc") is False + assert pbs_installer_supported_arch("mips") is False