Added instruction set details to unit test report.

kiyo-masui · Sep 27, 2024 · 385a6f3 · 385a6f3
1 parent 52aec3b
commit 385a6f3
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -0,0 +1,10 @@
+import pytest
+import bitshuffle
+
+def pytest_report_header(config):
+    sse2 = bitshuffle.using_SSE2()
+    avx2 = bitshuffle.using_AVX2()
+    avx512 = bitshuffle.using_AVX512()
+    neon = bitshuffle.using_NEON()
+    return f"SSE2 {sse2}; AVX2 {avx2}; AVX512 {avx512}; NEON {neon}"
+