J/PhP+<ddlZddlZddlZddlZddlZddlZddlZddlmZddl Z ddl m Z ddl m Z ddlmZddlmZmZGdde ZGd d e ZGd d e ZGd deZdZdZdZdZedkrejdSdS)N) get_context) TypingError)TestCase)resolve_dispatcher_from_strdumpsloadsceZdZdZejdZdZdZdZ dZ dZ dZ d Z d Zd Zd Zd ZdZdZdZdZdZdZdZdS)TestDispatcherPicklingc\ttjdzD]}||g|Ri|dS)N)rangepickleHIGHEST_PROTOCOL)selfmethargskwargsprotos Z/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/numba/tests/test_serialize.pyrun_with_protocolsz)TestDispatcherPickling.run_with_protocolssN62Q677 ) )E D ( ( ( ( ( ( ( ( ) )c#Kd}t|}|j}t|j||_ dV||_dS#||_wxYw)Ncpu)r targetdescrtype)rhwstrdispatcher_cls old_descrs rsimulate_fresh_targetz,TestDispatcherPickling.simulate_fresh_targetso4U;;". %ET.*D%E%Ee%L%L" 3 EEE*3N & & &N & 2 2 2 2s A Acfd}||tj||}5tj|}||ddddS#1swxYwYdS)Ncttr&ttrj|gRdS|dS)N) isinstancer issubclass Exception assertRaisesassertPreciseEqual)funcrexpected_resultrs r check_resultz7TestDispatcherPickling.check_call..check_result&sn?D11 F :: F!!/4?$??????''d _EEEEEr)rrr r )rrr(r)rr*picklednew_funcs` `` r check_callz!TestDispatcherPickling.check_call%s F F F F F F F  T,tU++  ' ' ) ) # #|G,,H L " " " # # # # # # # # # # # # # # # # # #s A,,A03A0cddlm}||j|dd||j|dddS)Nr ) add_with_sigr )333333?g@)serialize_usecasesr/rr-)rr/s rtest_call_with_sigz)TestDispatcherPickling.test_call_with_sig4sR444444 q&III q*MMMMMrcddlm}||j|dd||j|dd||j|dddS) Nr )add_without_sigr0r1@r3g333333@abc)abc)r4r7rr-)rr7s rtest_call_without_sigz,TestDispatcherPickling.test_call_without_sig:sn777777 !VLLL #zRRR %UUUUUrcddlm}||j|dd||j|tt t fdS)Nr ) add_nopythonr8r9)r4r?rr-robject)rr?s rtest_call_nopythonz)TestDispatcherPickling.test_call_nopythonAsb444444 sJOOO {VXXW]W_W_L`aaaaarcVddlm}||j|tddS)Nr )add_nopython_fail)r )r4rCrr-r)rrCs rtest_call_nopython_failz.TestDispatcherPickling.test_call_nopython_failGs8999999 1BKQWXXXXXrcLddlm}||j|dddS)Nr )get_global_objmodeg@)g@)r4rGrr-)rrGs rtest_call_objmode_with_globalz4TestDispatcherPickling.test_call_objmode_with_globalLs7:::::: 1CS&QQQQQrcbddlm}|d}||j|dddS)Nr closure)rD)r4rKrr-)rrKinners rtest_call_closurez(TestDispatcherPickling.test_call_closurePsC//////  6BBBBBrc \ddlm}|di|}||j|dddS)Nr )closure_with_globals@g@@)rR)r4rQrr-)rjit_argsrQrNs rcheck_call_closure_with_globalsz6TestDispatcherPickling.check_call_closure_with_globalsUsL<<<<<<$$55H55 VDDDDDrc2|ddS)NT)nopythonrVrs r'test_call_closure_with_globals_nopythonz>TestDispatcherPickling.test_call_closure_with_globals_nopythonZ ,,d,;;;;;rc2|ddS)NT)forceobjrYrZs r&test_call_closure_with_globals_objmodez=TestDispatcherPickling.test_call_closure_with_globals_objmode]r\rcbddlm}|d}||j|dddS)Nr )closure_calling_other_functionrRg&@)rTg@)r4rarr-)rrarNs r(test_call_closure_calling_other_functionz?TestDispatcherPickling.test_call_closure_calling_other_function`sFFFFFFF..s33 jIIIIIrcbddlm}|d}||j|dddS)Nr closure_calling_other_closurerRg @rS)r4rerr-)rrerNs r'test_call_closure_calling_other_closurez>TestDispatcherPickling.test_call_closure_calling_other_closureesFEEEEEE--c22 VDDDDDrcLddlm}||j|dddS)Nr )dyn_func$rL)r4rhrr-)rrhs rtest_call_dyn_funcz)TestDispatcherPickling.test_call_dyn_funcjs6000000 2tDDDDDrcLddlm}||j|dddS)Nr )dyn_func_objmoderirj)r4rmrr-)rrms rtest_call_dyn_func_objmodez1TestDispatcherPickling.test_call_dyn_func_objmodeos7888888 1A2tLLLLLrcbddlm}|d}||j||ddS)Nr )get_renamed_module)rq)r4rprr-)rrpexpecteds rtest_renamed_modulez*TestDispatcherPickling.test_renamed_moduletsT::::::&%c** 1C (& 2 2 2 2 2rcddlm}|d}tj|}djdit }t jtj d|gdS)z} Check that reconstructing doesn't depend on resources already instantiated in the original process. r rdrRzif 1: import pickle data = {pickled!r} func = pickle.loads(data) res = func(4.0) assert res == 8.0, res -cN) r4rerrformatlocals subprocessr-sys executable)rrer(r+codes rtest_other_processz)TestDispatcherPickling.test_other_process|s FEEEEE,,S11,t$$## ## s~tT:;;;;;rcddlm}|d}tj|}|d}tj|}tj|}tj|}tj|}|||||ddd|||ddd |||||ddd |d }|d d tj|}~tj tj|}tj|}|||||ddd |||ddddS)z Check that deserializing the same function multiple times re-uses the same dispatcher object. Note that "same function" is intentionally under-specified. r rJr0rLrDrM r2 *+ N) r4rKrrr assertIs assertEqualdisable_compile assertIsNotgccollect) rrKr(r+func2pickled2fghs r test_reusez!TestDispatcherPickling.test_reuses 0/////wqzz,t$$ <&& L ! ! L ! ! L " " a 1a"%%%  1a"%%% A 1a"%%%wqzz R ,t$$  L ! ! L ! ! a 1a"%%%  1a"%%%%%rcLd}tjtjd|gdS)zN The imp module was deprecated in v3.4 in favour of importlib aif 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 ruN)ryr-rzr{)rr|s rtest_imp_deprecationz+TestDispatcherPickling.test_imp_deprecations+  s~tT:;;;;;rN)__name__ __module__ __qualname__r contextlibcontextmanagerr r-r5r=rArErHrOrVr[r_rbrfrkrnrsr}rrrvrrr r sr))) 3 3 3 # # #NNN VVVbbb YYY RRRCCC EEE <<<<<<JJJ EEE EEE MMM 222<<<$$&$&$&L<<<<.KlassNrrrclassvarrvrrKlassrHHHrrcd_dSNd)rrsrmutatorzKTestCloudPickleIssues.test_dynamic_class_reset_on_unpickle..mutators ENNNrc>jddSr)rr)rrsrcheckzITestCloudPickleIssues.test_dynamic_class_reset_on_unpickle..checks!   U^S 1 1 1 1 1rr)rrrsavedrs` @r$test_dynamic_class_reset_on_unpicklez:TestCloudPickleIssues.test_dynamic_class_reset_on_unpickles         ! ! ! ! ! 2 2 2 2 2 2e    e   e  r__main__z*Test cannot run as when module is __main__ctd}|t}||d||jddS)Nspawn)target<timeoutr)rProcess"check_main_class_reset_on_unpicklestartjoinrexitcode)rmpprocs r!test_main_class_reset_on_unpicklez7TestCloudPickleIssues.test_main_class_reset_on_unpicklesd ! !zz!CzDD  "  *****rc Gdd}t|}td}|t|f}||d||jddS)NceZdZdZdS)RTestCloudPickleIssues.test_dynamic_class_reset_on_unpickle_new_proc..KlassNrrvrrrrrrrr)rrrrr)rrr!check_unpickle_dyn_class_new_procrrrr)rrrrrs r-test_dynamic_class_reset_on_unpickle_new_proczCTestCloudPickleIssues.test_dynamic_class_reset_on_unpickle_new_procs        e  ! !zz!B%zRR  "  *****rcttjt}||ddS)N)rr)numbanjit issue_7356r)rcfuncs rtest_dynamic_class_issue_7356z3TestCloudPickleIssues.test_dynamic_class_issue_7356 s3 :&& *-----rN) rrr__doc___numba_parallel_test_runittestskipIfrrrrvrrrrs!(X_X+ACC++CC+ + + +.....rrceZdZdZdS)DynClassN)rrrr;rvrrrrs AAArrctjd5dt_tj}dddn #1swxYwYtjd5tj}dddn #1swxYwY||fS)Nintp)beforer)after)robjmoderr;)rrs rrrs f % % %  V $ $ $  5=s;?? A44A8;A8ctjdd}|d}|jdksJ|jdksJt |}d|_t ||dS)Nz"numba.tests.cloudpickle_main_classr)run_namerr)runpy run_modulerrr_check_dyn_class)glbsrrs rrrsy  ,   D ME  z ) ) ) ) >S %LLEENUE"""""rclt|}|jdksJd|_t||dSr)r rr)rrs rrr0s= %LLE >S ENUE"""""rcfd}|t|}|urtd|t|}|urtd|dS)Nc<jdkrtddS)NrzCheck failed. Klass reset.)rAssertionErrorrsrrz_check_dyn_class..check:s' >S  !=>> > ! rzExpected reuse)r r)rrrloadeds` rrr9s????? EGGG 5\\F U-... EGGG 5\\F U-... EGGGGGrr)rrrrryrzrmultiprocessingrrnumba.core.errorsrnumba.tests.supportrnumba.core.target_extensionrnumba.cloudpicklerr r rrr@rrrrrrmainrvrrrs  '''''' ))))))((((((CCCCCC********v<v<v<v<v