Fix module names that don't have a .

asmeurer · asmeurer · commit 090c7c6f75cf · 2020-09-11T15:46:16.000-06:00
diff --git a/array_api_tests/_array_module.py b/array_api_tests/_array_module.py
@@ -9,7 +9,9 @@
 if array_module is None:
     if 'ARRAY_API_TESTS_MODULE' in os.environ:
         mod_name = os.environ['ARRAY_API_TESTS_MODULE']
-        _module, _sub = mod_name.split('.', 1)
+        _module, _sub = mod_name, None
+        if '.' in mod_name:
+            _module, _sub = mod_name.split('.', 1)
         mod = import_module(_module)
         if _sub:
             mod = getattr(mod, _sub)
