Start file for parameterizing the array module

asmeurer · asmeurer · commit d95abdf20f0e · 2020-09-04T14:25:04.000-06:00
diff --git a/array_api_tests/_array_module.py b/array_api_tests/_array_module.py
@@ -0,0 +1,22 @@
+import os
+from importlib import import_module
+
+# Replace this with a specific array module to test it, for example,
+#
+# import numpy as array_module
+array_module = None
+
+if array_module is None:
+    if 'ARRAY_API_TESTS_MODULE' in os.environ:
+        mod_name = os.environ['ARRAY_API_TESTS_MODULE']
+        mod = import_module(mod_name)
+    else:
+        raise RuntimeError("No array module specified. Either edit _array_module.py or set the ARRAY_API_TESTS_MODULE environment variable")
+else:
+    mod = array_module
+
+# Names from the spec. This is what should actually be imported from this
+# file.
+
+array = mod.array
+dtype = mod.dtype
