tests: Fix AVX512 detection on Numpy>=1.26 (#2932)

The output format of show_config() changed, but the library now provides
and option to return the values in a dictionary.

Signed-off-by: Jan Vesely <[email protected]>

conftest.py

@@ -235,11 +235,25 @@ def mech_wrapper(x):
 
 @pytest.helpers.register
 def numpy_uses_avx512():
-    out = io.StringIO()
-    with contextlib.redirect_stdout(out):
-        np.show_config()
 
-    return re.search('  found = .*AVX512.*', out.getvalue()) is not None
+    try:
+        # numpy >= 1.26 can return config info in a dictionary
+        config = np.show_config(mode="dicts")
+
+    except TypeError:
+        # Numpy >=1.21 < 1.26 doesn't support 'mode' argument and
+        # prints CPU extensions in one line per category:
+        # baseline = ...
+        # found = ...
+        # not found = ...
+        out = io.StringIO()
+
+        with contextlib.redirect_stdout(out):
+            np.show_config()
+
+        return re.search('  found = .*AVX512.*', out.getvalue()) is not None
+    else:
+        return any(ext.startswith("AVX512") for ext in config['SIMD Extensions']['found'])
 
 @pytest.helpers.register
 def expand_np_ndarray(arr):