← Back
Editing: test_source.py
#!/usr/bin/env python # # Author: Mike McKerns (mmckerns @caltech and @uqfoundation) # Copyright (c) 2008-2016 California Institute of Technology. # Copyright (c) 2016-2023 The Uncertainty Quantification Foundation. # License: 3-clause BSD. The full license text is available at: # - https://github.com/uqfoundation/dill/blob/master/LICENSE from dill.source import getsource, getname, _wrap, likely_import from dill.source import getimportable from dill._dill import IS_PYPY import sys PY310b = 0x30a00b1 f = lambda x: x**2 def g(x): return f(x) - x def h(x): def g(x): return x return g(x) - x class Foo(object): def bar(self, x): return x*x+x _foo = Foo() def add(x,y): return x+y # yes, same as 'f', but things are tricky when it comes to pointers squared = lambda x:x**2 class Bar: pass _bar = Bar() # inspect.getsourcelines # dill.source.getblocks def test_getsource(): assert getsource(f) == 'f = lambda x: x**2\n' assert getsource(g) == 'def g(x): return f(x) - x\n' assert getsource(h) == 'def h(x):\n def g(x): return x\n return g(x) - x\n' assert getname(f) == 'f' assert getname(g) == 'g' assert getname(h) == 'h' assert _wrap(f)(4) == 16 assert _wrap(g)(4) == 12 assert _wrap(h)(4) == 0 assert getname(Foo) == 'Foo' assert getname(Bar) == 'Bar' assert getsource(Bar) == 'class Bar:\n pass\n' assert getsource(Foo) == 'class Foo(object):\n def bar(self, x):\n return x*x+x\n' #XXX: add getsource for _foo, _bar # test itself def test_itself(): assert likely_import(likely_import)=='from dill.source import likely_import\n' # builtin functions and objects def test_builtin(): assert likely_import(pow) == 'pow\n' assert likely_import(100) == '100\n' assert likely_import(True) == 'True\n' assert likely_import(pow, explicit=True) == 'from builtins import pow\n' assert likely_import(100, explicit=True) == '100\n' assert likely_import(True, explicit=True) == 'True\n' # this is kinda BS... you can't import a None assert likely_import(None) == 'None\n' assert likely_import(None, explicit=True) == 'None\n' # other imported functions def test_imported(): from math import sin assert likely_import(sin) == 'from math import sin\n' # interactively defined functions def test_dynamic(): assert likely_import(add) == 'from %s import add\n' % __name__ # interactive lambdas assert likely_import(squared) == 'from %s import squared\n' % __name__ # classes and class instances def test_classes(): from io import BytesIO as StringIO y = "from _io import BytesIO\n" x = y if (IS_PYPY or sys.hexversion >= PY310b) else "from io import BytesIO\n" s = StringIO() assert likely_import(StringIO) == x assert likely_import(s) == y # interactively defined classes and class instances assert likely_import(Foo) == 'from %s import Foo\n' % __name__ assert likely_import(_foo) == 'from %s import Foo\n' % __name__ # test getimportable def test_importable(): assert getimportable(add) == 'from %s import add\n' % __name__ assert getimportable(squared) == 'from %s import squared\n' % __name__ assert getimportable(Foo) == 'from %s import Foo\n' % __name__ assert getimportable(Foo.bar) == 'from %s import bar\n' % __name__ assert getimportable(_foo.bar) == 'from %s import bar\n' % __name__ assert getimportable(None) == 'None\n' assert getimportable(100) == '100\n' assert getimportable(add, byname=False) == 'def add(x,y):\n return x+y\n' assert getimportable(squared, byname=False) == 'squared = lambda x:x**2\n' assert getimportable(None, byname=False) == 'None\n' assert getimportable(Bar, byname=False) == 'class Bar:\n pass\n' assert getimportable(Foo, byname=False) == 'class Foo(object):\n def bar(self, x):\n return x*x+x\n' assert getimportable(Foo.bar, byname=False) == 'def bar(self, x):\n return x*x+x\n' assert getimportable(Foo.bar, byname=True) == 'from %s import bar\n' % __name__ assert getimportable(Foo.bar, alias='memo', byname=True) == 'from %s import bar as memo\n' % __name__ assert getimportable(Foo, alias='memo', byname=True) == 'from %s import Foo as memo\n' % __name__ assert getimportable(squared, alias='memo', byname=True) == 'from %s import squared as memo\n' % __name__ assert getimportable(squared, alias='memo', byname=False) == 'memo = squared = lambda x:x**2\n' assert getimportable(add, alias='memo', byname=False) == 'def add(x,y):\n return x+y\n\nmemo = add\n' assert getimportable(None, alias='memo', byname=False) == 'memo = None\n' assert getimportable(100, alias='memo', byname=False) == 'memo = 100\n' assert getimportable(add, explicit=True) == 'from %s import add\n' % __name__ assert getimportable(squared, explicit=True) == 'from %s import squared\n' % __name__ assert getimportable(Foo, explicit=True) == 'from %s import Foo\n' % __name__ assert getimportable(Foo.bar, explicit=True) == 'from %s import bar\n' % __name__ assert getimportable(_foo.bar, explicit=True) == 'from %s import bar\n' % __name__ assert getimportable(None, explicit=True) == 'None\n' assert getimportable(100, explicit=True) == '100\n' def test_numpy(): try: from numpy import array x = array([1,2,3]) assert getimportable(x) == 'from numpy import array\narray([1, 2, 3])\n' assert getimportable(array) == 'from %s import array\n' % array.__module__ assert getimportable(x, byname=False) == 'from numpy import array\narray([1, 2, 3])\n' assert getimportable(array, byname=False) == 'from %s import array\n' % array.__module__ except ImportError: pass #NOTE: if before likely_import(pow), will cause pow to throw AssertionError def test_foo(): assert getimportable(_foo, byname=False).startswith("import dill\nclass Foo(object):\n def bar(self, x):\n return x*x+x\ndill.loads(") if __name__ == '__main__': test_getsource() test_itself() test_builtin() test_imported() test_dynamic() test_classes() test_importable() test_numpy() test_foo()
Save File
Cancel