def test_load_module_py(): fn = _py_file() descr = ('.py', 'U', imp.PY_SOURCE) f = open(fn, 'U') mod = imp.load_module('test_imp_extra_AUTO1', f, fn, descr) f.close() assert mod.MARKER == 42 import test_imp_extra_AUTO1 assert mod is test_imp_extra_AUTO1
def test_load_module_pyc(): fn = _pyc_file() try: descr = ('.pyc', 'rb', imp.PY_COMPILED) f = open(fn, 'rb') mod = imp.load_module('test_imp_extra_AUTO2', f, fn, descr) f.close() assert mod.marker == 42 import test_imp_extra_AUTO2 assert mod is test_imp_extra_AUTO2 finally: os.unlink(fn)