def test_filename_simple(self):
self.w("foo_extended.py", """
answer = 42
""")
#
myglobals = {}
extend_module_maybe(myglobals, filename=self.tmpdir.join('foo.py'))
assert myglobals['answer'] == 42
def test_modname_simple(self, monkeypatch):
monkeypatch.syspath_prepend(self.tmpdir)
self.w("foo_extended.py", """
answer = 42
""")
#
myglobals = {}
extend_module_maybe(myglobals, modname='foo')
assert myglobals['answer'] == 42
def test_filename_package(self):
mypackage = self.tmpdir.join('mypackage').ensure(dir=True)
mypackage.join('__init__.py').write('')
self.w('mypackage/foo_extended.py', """
answer = 42
""")
#
myglobals = {}
extend_module_maybe(myglobals, filename=self.tmpdir.join('mypackage', 'foo.py'))
assert myglobals['answer'] == 42
def test_getsource(self):
self.w("foo_extended.py", """
def foo(): return 42
""")
#
myglobals = {}
extend_module_maybe(myglobals, filename=self.tmpdir.join('foo.py'))
foo = myglobals['foo']
assert foo() == 42
src = inspect.getsource(foo)
assert src.strip() == 'def foo(): return 42'
def test_modname_package(self, monkeypatch):
monkeypatch.syspath_prepend(self.tmpdir)
mypackage = self.tmpdir.join('mypackage').ensure(dir=True)
mypackage.join('__init__.py').write('')
self.w('mypackage/foo_extended.py', """
answer = 42
""")
#
myglobals = {}
extend_module_maybe(myglobals, modname='mypackage.foo')
assert myglobals['answer'] == 42
def test_modname_dont_exist(self, monkeypatch):
monkeypatch.syspath_prepend(self.tmpdir)
myglobals = {}
extend_module_maybe(myglobals, modname='foo')
assert myglobals == {}
def test_filename_dont_exist(self):
myglobals = {}
extend_module_maybe(myglobals, filename=self.tmpdir.join('foo.py'))
assert myglobals == {}
