σ \ K]c@`sΫddlmZmZmZddlZddlZddlZddlZddlZddl m Z ddl m Z ddlmZddlmZmZddlTd efd „ƒYZed krΧe jƒndS( i(tprint_functiontabsolute_importtdivisionN(tunittest_support(t TypingError(tregistryi(tTestCasettag(t*tTestDispatcherPicklingcB`s eZd„Zejd„ƒZd„Zedƒd„ƒZedƒd„ƒZ edƒd„ƒZ d„Z d„Z d „Z d „Zd „Zd „Zd „Zd„Zd„Zd„Zd„Zd„Zedƒd„ƒZedƒd„ƒZd„ZRS(cO`s2x+ttjdƒD]}||||ŽqWdS(Ni(trangetpickletHIGHEST_PROTOCOL(tselftmethtargstkwargstproto((s9lib/python2.7/site-packages/numba/tests/test_serialize.pytrun_with_protocolsscc`sEtjd}|j}t|jƒƒ|_z dVWd||_XdS(Ntcpu(Rtdispatcher_registryt targetdescrttype(R tdispatcher_clst old_descr((s9lib/python2.7/site-packages/numba/tests/test_serialize.pytsimulate_fresh_targets    c`sa‡‡‡fd†}||ƒtj||ƒ}ˆjƒtj|ƒ}||ƒWdQXdS(Nc`sNtˆtƒr4tˆtƒr4ˆjˆ|ˆŒnˆj|ˆŒˆƒdS(N(t isinstanceRt issubclasst Exceptiont assertRaisestassertPreciseEqual(tfunc(Rtexpected_resultR (s9lib/python2.7/site-packages/numba/tests/test_serialize.pyt check_result$s(R tdumpsRtloads(R RRR RR!tpickledtnew_func((RR R s9lib/python2.7/site-packages/numba/tests/test_serialize.pyt check_call#s   t importantcC`s6|j|jtddƒ|j|jtddƒdS(Niiig333333σ?gΝΜΜΜΜΜ@(ii(g333333σ?gΝΜΜΜΜΜ@(RR&t add_with_sig(R ((s9lib/python2.7/site-packages/numba/tests/test_serialize.pyttest_call_with_sig2scC`sO|j|jtdd ƒ|j|jtdd ƒ|j|jtdd ƒdS( Niiig@g333333σ?g333333@tabctatbc(ii(g333333σ?g333333@(R+R,(RR&tadd_without_sig(R ((s9lib/python2.7/site-packages/numba/tests/test_serialize.pyttest_call_without_sig8scC`sB|j|jtddƒ|j|jtttƒtƒfƒdS(Ng@g333333σ?g333333@(g333333σ?g333333@(RR&t add_nopythonRtobject(R ((s9lib/python2.7/site-packages/numba/tests/test_serialize.pyttest_call_nopython?scC`s|j|jttdƒdS(Nii(ii(RR&tadd_nopython_failR(R ((s9lib/python2.7/site-packages/numba/tests/test_serialize.pyttest_call_nopython_failEscC`s|j|jtddƒdS(Ng@g@(g@(RR&tget_global_objmode(R ((s9lib/python2.7/site-packages/numba/tests/test_serialize.pyttest_call_objmode_with_globalIscC`s)tdƒ}|j|j|ddƒdS(Niiii(ii(tclosureRR&(R tinner((s9lib/python2.7/site-packages/numba/tests/test_serialize.pyttest_call_closureLs cK`s,td|}|j|j|ddƒdS(Ng@g@g@(g@(tclosure_with_globalsRR&(R tjit_argsR7((s9lib/python2.7/site-packages/numba/tests/test_serialize.pytcheck_call_closure_with_globalsPscC`s|jdtƒdS(Ntnopython(R;tTrue(R ((s9lib/python2.7/site-packages/numba/tests/test_serialize.pyt'test_call_closure_with_globals_nopythonTscC`s|jdtƒdS(Ntforceobj(R;R=(R ((s9lib/python2.7/site-packages/numba/tests/test_serialize.pyt&test_call_closure_with_globals_objmodeWscC`s)tdƒ}|j|j|ddƒdS(Ng@g&@g@g@(g@g@(tclosure_calling_other_functionRR&(R R7((s9lib/python2.7/site-packages/numba/tests/test_serialize.pyt(test_call_closure_calling_other_functionZs cC`s)tdƒ}|j|j|ddƒdS(Ng@g @g@(g@(tclosure_calling_other_closureRR&(R R7((s9lib/python2.7/site-packages/numba/tests/test_serialize.pyt'test_call_closure_calling_other_closure^s cC`s|j|jtddƒdS(Ni$i(i(RR&tdyn_func(R ((s9lib/python2.7/site-packages/numba/tests/test_serialize.pyttest_call_dyn_funcbscC`s|j|jtddƒdS(Ni$i(i(RR&tdyn_func_objmode(R ((s9lib/python2.7/site-packages/numba/tests/test_serialize.pyttest_call_dyn_func_objmodefscC`s)tdƒ}|j|jt|dƒdS(Ng(g(tget_renamed_moduleRR&(R texpected((s9lib/python2.7/site-packages/numba/tests/test_serialize.pyttest_renamed_modulejs cC`s6|j|jtddƒ|j|jtddƒdS( Ni.iiyπ?i(iiy@π?(yπ?i(RR&t generated_add(R ((s9lib/python2.7/site-packages/numba/tests/test_serialize.pyttest_call_generatedqs cC`sJtdƒ}tj|ƒ}djtƒ}tjtjd|gƒdS(s} Check that reconstructing doesn't depend on resources already instantiated in the original process. g@s±if 1: import pickle data = {pickled!r} func = pickle.loads(data) res = func(4.0) assert res == 8.0, res s-cN( RCR R"tformattlocalst subprocessR&tsyst executable(R RR$tcode((s9lib/python2.7/site-packages/numba/tests/test_serialize.pyttest_other_processws cC`s{tdƒ}tj|ƒ}tdƒ}tj|ƒ}tj|ƒ}tj|ƒ}tj|ƒ}|j||ƒ|j|ddƒdƒ|jƒ|j|ddƒdƒ|j||ƒ|j|ddƒdƒtdƒ}|d d ƒtj|ƒ}~tj ƒtj|ƒ}tj|ƒ}|j||ƒ|j|ddƒd ƒ|jƒ|j|ddƒd ƒd S(sΌ Check that deserializing the same function multiple times re-uses the same dispatcher object. Note that "same function" is intentionally under-specified. iiiii ii ii*i+i i N( R6R R"R#tassertIst assertEqualtdisable_compilet assertIsNottgctcollect(R RR$tfunc2tpickled2tftgth((s9lib/python2.7/site-packages/numba/tests/test_serialize.pyt test_reuse‰s0       cC`s#d}tjtjd|gƒdS(sN The imp module was deprecated in v3.4 in favour of importlib sif 1: import pickle import warnings with warnings.catch_warnings(record=True) as w: warnings.simplefilter('always', DeprecationWarning) from numba import njit @njit def foo(x): return x + 1 foo(1) serialized_foo = pickle.dumps(foo) for x in w: if 'serialize.py' in x.filename: assert "the imp module is deprecated" not in x.msg s-cN(RPR&RQRR(R RS((s9lib/python2.7/site-packages/numba/tests/test_serialize.pyttest_imp_deprecation―s(t__name__t __module__Rt contextlibtcontextmanagerRR&RR)R.R1R3R5R8R;R>R@RBRDRFRHRKRMRTR`Ra(((s9lib/python2.7/site-packages/numba/tests/test_serialize.pyR s*              &t__main__(t __future__RRRRdRYR RPRQtnumbaRtunittestt numba.errorsRt numba.targetsRtsupportRRtserialize_usecasesR Rbtmain(((s9lib/python2.7/site-packages/numba/tests/test_serialize.pyts      ΄