tests: Add tests for overriding builtins.__import__.

1 files changed, 17 insertions, 0 deletions

diff --git a/tests/import/import_override.py b/tests/import/import_override.py
new file mode 100644
index 000000000..6fe99009e
--- /dev/null
+++ b/tests/import/import_override.py
@@ -0,0 +1,17 @@
+# test overriding __import__ combined with importing from the filesystem
+
+def custom_import(name, globals, locals, fromlist, level):
+    print('import', name, fromlist, level)
+    class M:
+        var = 456
+    return M
+
+orig_import = __import__
+try:
+    __import__("builtins").__import__ = custom_import
+except AttributeError:
+    print("SKIP")
+    raise SystemExit
+
+# import1a will be done via normal import which will import1b via our custom import
+orig_import('import1a')