diff options
| -rw-r--r-- | numpy/core/tests/test_cpu_features.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/numpy/core/tests/test_cpu_features.py b/numpy/core/tests/test_cpu_features.py index 2b2272769..706cf7a7e 100644 --- a/numpy/core/tests/test_cpu_features.py +++ b/numpy/core/tests/test_cpu_features.py @@ -146,14 +146,17 @@ class Test_POWER_Features(AbstractTest): def load_flags(self): self.load_flags_auxv() + is_zarch = re.match("^(s390x)", machine, re.IGNORECASE) -@pytest.mark.skipif(not is_linux or not is_zarch, reason="Only for Linux and IBM Z") +@pytest.mark.skipif(not is_linux or not is_zarch, + reason="Only for Linux and IBM Z") class Test_ZARCH_Features(AbstractTest): features = ["VX", "VXE", "VXE2"] def load_flags(self): self.load_flags_auxv() + is_arm = re.match("^(arm|aarch64)", machine, re.IGNORECASE) @pytest.mark.skipif(not is_linux or not is_arm, reason="Only for Linux and ARM") class Test_ARM_Features(AbstractTest): |