1from zipped import zipped_contents
2try:
3    code = zipped_contents.call_in_zip.__code__
4except AttributeError:
5    code = zipped_contents.call_in_zip.func_code
6assert 'myzip.zip' in code.co_filename
7zipped_contents.call_in_zip()
8
9from zipped2 import zipped_contents2
10try:
11    code = zipped_contents2.call_in_zip2.__code__
12except AttributeError:
13    code = zipped_contents2.call_in_zip2.func_code
14assert 'myzip2.egg!' in code.co_filename
15zipped_contents2.call_in_zip2()
16
17print('TEST SUCEEDED!')
18