def run_test(self, source): with source_util.create_modules(self.module_name) as mapping: with open(mapping[self.module_name], u'wb') as file: file.write(source) loader = _bootstrap._SourceFileLoader(self.module_name, mapping[self.module_name]) return loader.load_module(self.module_name)
def test_bad_syntax(self): with source_util.create_modules(u'_temp') as mapping: with open(mapping[u'_temp'], u'w') as file: file.write(u'=') loader = _bootstrap._SourceFileLoader(u'_temp', mapping[u'_temp']) with self.assertRaises(SyntaxError): loader.load_module(u'_temp') self.assertTrue(u'_temp' not in sys.modules)
def test_module(self): with source_util.create_modules(u'_temp') as mapping: loader = _bootstrap._SourceFileLoader(u'_temp', mapping[u'_temp']) module = loader.load_module(u'_temp') self.assertTrue(u'_temp' in sys.modules) check = {u'__name__': u'_temp', u'__file__': mapping[u'_temp'], u'__package__': u''} for attr, value in check.items(): self.assertEqual(getattr(module, attr), value)
def run_test(self, line_ending): module_name = u'_temp' source_lines = ["a = 42", "b = -13", ''] source = line_ending.join(source_lines) with source_util.create_modules(module_name) as mapping: with open(mapping[module_name], u'wb') as file: file.write(source) loader = _bootstrap._SourceFileLoader(module_name, mapping[module_name]) return loader.load_module(module_name)
def test_state_after_failure(self): # A failed reload should leave the original module intact. attributes = (u'__file__', u'__path__', u'__package__') value = u'<test>' name = u'_temp' with source_util.create_modules(name) as mapping: orig_module = imp.new_module(name) for attr in attributes: setattr(orig_module, attr, value) with open(mapping[name], u'w') as file: file.write(u'+++ bad syntax +++') loader = _bootstrap._SourceFileLoader(u'_temp', mapping[u'_temp']) with self.assertRaises(SyntaxError): loader.load_module(name) for attr in attributes: self.assertEqual(getattr(orig_module, attr), value)
def test_file_from_empty_string_dir(self): # Loading a module found from an empty string entry on sys.path should # not only work, but keep all attributes relative. file_path = u'_temp.py' with open(file_path, u'w') as file: file.write(u"# test file for importlib_full") try: with util.uncache(u'_temp'): loader = _bootstrap._SourceFileLoader(u'_temp', file_path) mod = loader.load_module(u'_temp') self.assertEqual(file_path, mod.__file__) self.assertEqual(imp.cache_from_source(file_path), mod.__cached__) finally: os.unlink(file_path) pycache = os.path.dirname(imp.cache_from_source(file_path)) shutil.rmtree(pycache)
def test_module_reuse(self): with source_util.create_modules(u'_temp') as mapping: loader = _bootstrap._SourceFileLoader(u'_temp', mapping[u'_temp']) module = loader.load_module(u'_temp') module_id = id(module) module_dict_id = id(module.__dict__) with open(mapping[u'_temp'], u'w') as file: file.write(u"testing_var = 42\n") # For filesystems where the mtime is only to a second granularity, # everything that has happened above can be too fast; # force an mtime on the source that is guaranteed to be different # than the original mtime. loader.path_mtime = self.fake_mtime(loader.path_mtime) module = loader.load_module(u'_temp') self.assertTrue(u'testing_var' in module.__dict__, u"'testing_var' not in " u"{0}".format(list(module.__dict__.keys()))) self.assertEqual(module, sys.modules[u'_temp']) self.assertEqual(id(module), module_id) self.assertEqual(id(module.__dict__), module_dict_id)