tests/import/import_broken.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32

import sys, pkg

# Modules we import are usually added to sys.modules.
print("pkg" in sys.modules)

try:
    from broken.zerodiv import x
except Exception as e:
    print(e.__class__.__name__)

# The broken module we tried to import should not be in sys.modules.
print("broken.zerodiv" in sys.modules)

# If we try to import the module again, the code should
# run again and we should get the same error.
try:
    from broken.zerodiv import x
except Exception as e:
    print(e.__class__.__name__)

# Import a module that successfully imports some other modules
# before importing the problematic module.
try:
    import broken.pkg2_and_zerodiv
except ZeroDivisionError:
    pass

print("pkg2" in sys.modules)
print("pkg2.mod1" in sys.modules)
print("pkg2.mod2" in sys.modules)
print("broken.zerodiv" in sys.modules)
print("broken.pkg2_and_zerodiv" in sys.modules)