Make the array API submodule not break the test suite

The warning is issued on import, which otherwise breaks pytest collection. If
we manually import early and ignore the warning, any further imports of the
module won't issue the warning again, due to the way Python caches imports.

1 files changed, 10 insertions, 3 deletions

diff --git a/numpy/_pytesttester.py b/numpy/_pytesttester.py
index acfaa1ca5..bfcbd4f1f 100644
--- a/numpy/_pytesttester.py
+++ b/numpy/_pytesttester.py
@@ -137,13 +137,20 @@ class PytestTester:
         # offset verbosity. The "-q" cancels a "-v".
         pytest_args += ["-q"]
 
-        # Filter out distutils cpu warnings (could be localized to
-        # distutils tests). ASV has problems with top level import,
-        # so fetch module for suppression here.
         with warnings.catch_warnings():
             warnings.simplefilter("always")
+            # Filter out distutils cpu warnings (could be localized to
+            # distutils tests). ASV has problems with top level import,
+            # so fetch module for suppression here.
             from numpy.distutils import cpuinfo
 
+            # Ignore the warning from importing the array_api submodule. This
+            # warning is done on import, so it would break pytest collection,
+            # but importing it early here prevents the warning from being
+            # issued when it imported again.
+            warnings.simplefilter("ignore")
+            import numpy.array_api
+
         # Filter out annoying import messages. Want these in both develop and
         # release mode.
         pytest_args += [