o V۷i@s*ddlmZmZmZmZddlmZddlmZddlZdZ ddZ dd Z d d Z Gd d d e ZeZddZddZGdddZeZddZddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zed'kreeeeeeeeeedSdS)() getsourcegetname_wrap getimport) importable)IS_PYPYNi cC|dSNxr r L/home/ubuntu/vllm_env/lib/python3.10/site-packages/dill/tests/test_source.pyrcCs t||SN)fr r r rgs rcCsdd}|||S)NcSs|Srr r r r rrsz h..gr )r rr r rhs rc@seZdZddZdS)FoocCs |||Srr )selfr r r rbars zFoo.barN)__name__ __module__ __qualname__rr r r rrs rcCs||Srr )r yr r raddsrcCrr r r r r rr rc@s eZdZdS)BarN)rrrr r r rr"srcCsttdksJttdksJttdksJttdks Jttdks(Jttdks0Jttddks:Jttdd ksDJttdd ksNJttd ksVJttd ks^Jttd ksfJttdksnJdS)Nzf = lambda x: x**2 zdef g(x): return f(x) - x z1def h(x): def g(x): return x return g(x) - x rrr rrrclass Bar: pass 8class Foo(object): def bar(self, x): return x*x+x )rrrrrrrrr r r rtest_getsource'sr#cCsttdksJdS)Nz"from dill.source import getimport )rr r r r test_itself9sr$cCsttdksJtddksJtddksJttdddks"Jtddddks,Jtddddks6Jtddks>JtddddksHJdS) Nzpow d100 TzTrue )builtinzfrom builtins import pow None )rpowr r r r test_builtin=sr*cCs ddlm}t|dksJdS)Nrsinzfrom math import sin )mathr,rr+r r r test_importedJs r.cCs,ttdtks JttdtksJdS)Nfrom %s import add from %s import squared )rrrsquaredr r r r test_dynamicOsr2cCsxddlm}d}tstjtkr|nd}|}t||ksJt||ks&Jttdtks0Jtt dtks:JdS)Nr)BytesIOzfrom _io import BytesIO zfrom io import BytesIO from %s import Foo ) ior3rsys hexversionPY310brrr_foo)StringIOrr sr r r test_classesUs r<cCsttdddtks JttdddtksJttdddtks$Jttjdddtks1Jttjdddtks>JtddddksHJtdddd ksRJttd dd ks\Jttd dd ksfJtdd ddkspJttd dd kszJttd ddksJttjd ddksJttjdddtksJttjddddtksJttddddtksJttddddtksJttdd ddksJttdd ddksJtddd ddksJtddd ddksJttd dddtksJttd dddtks Jttd dddtksJttjd dddtks(Jttjd dddtks7Jtdd dddksCJtdd ddd ksOJdS)NFsourcer/r0r4zfrom %s import bar r(r%r&Tzdef add(x,y): return x+y zsquared = lambda x:x**2 r!r"z!def bar(self, x): return x*x+x memo)aliasr>zfrom %s import bar as memo zfrom %s import Foo as memo zfrom %s import squared as memo zmemo = squared = lambda x:x**2 z'def add(x,y): return x+y memo = add z memo = None z memo = 100 )r'r>)rrrr1rrr9rr r r rtest_importablecs8rAcCszddl}|j}|gd}t|dddksJt|ddd|jks%Jt|dddks/Jt|ddd|jksr r r test_safes   r^__main__) dill.sourcerrrrr dill._dillrr6r8rrrobjectrr9rr1r_barr#r$r*r.r2r<rArTrVr^rr r r rsH