summaryrefslogtreecommitdiff
path: root/tests/import/import_override.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/import/import_override.py')
-rw-r--r--tests/import/import_override.py8
1 files changed, 6 insertions, 2 deletions
diff --git a/tests/import/import_override.py b/tests/import/import_override.py
index 6fe99009e..029ebe54c 100644
--- a/tests/import/import_override.py
+++ b/tests/import/import_override.py
@@ -1,11 +1,15 @@
# test overriding __import__ combined with importing from the filesystem
+
def custom_import(name, globals, locals, fromlist, level):
- print('import', name, fromlist, level)
+ print("import", name, fromlist, level)
+
class M:
var = 456
+
return M
+
orig_import = __import__
try:
__import__("builtins").__import__ = custom_import
@@ -14,4 +18,4 @@ except AttributeError:
raise SystemExit
# import1a will be done via normal import which will import1b via our custom import
-orig_import('import1a')
+orig_import("import1a")