iP+"ddlZddlZddlZddlZddlZddlZddlZddlmZddl Z ddl m Z ddl m Z ddlmZddlmZmZGdde ZGd d e ZGd d e ZGd deZdZdZdZdZedk(rej:yy)N) get_context) TypingError)TestCase)resolve_dispatcher_from_strdumpsloadsceZdZdZej dZdZdZdZ dZ dZ dZ d Z d Zd Zd Zd ZdZdZdZdZdZdZdZy)TestDispatcherPicklingc^ttjdzD]}||g|i|y)N)rangepickleHIGHEST_PROTOCOL)selfmethargskwargsprotos o/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/numba/tests/test_serialize.pyrun_with_protocolsz)TestDispatcherPickling.run_with_protocolss.622Q67E  ( ( (8c#Kd}t|}|j}t|j||_ d||_y#||_wxYww)Ncpu)r targetdescrtype)rhwstrdispatcher_cls old_descrs rsimulate_fresh_targetz,TestDispatcherPickling.simulate_fresh_targetsT4U;".. %ET.*D*D%Ee%L" 3 *3N &N &s:AA A AAcfd}||tj||}j5tj|}||dddy#1swYyxYw)Ncttr%ttrj|gyj |y)N) isinstancer issubclass Exception assertRaisesassertPreciseEqual)funcrexpected_resultrs r check_resultz7TestDispatcherPickling.check_call..check_result&sB?D1 :!!!/4?$?''d _Er)rrr r )rrr(r)rr*picklednew_funcs` `` r check_callz!TestDispatcherPickling.check_call%sQ F T,,tU+  ' ' )||G,H  "* ) )s A  A)cddlm}|j|j|dd|j|j|ddy)Nr ) add_with_sigr )333333?g@)serialize_usecasesr/rr-)rr/s rtest_call_with_sigz)TestDispatcherPickling.test_call_with_sig4s54 q&I q*Mrcddlm}|j|j|dd|j|j|dd|j|j|ddy) Nr )add_without_sigr0r1@r3g333333@abc)abc)r4r7rr-)rr7s rtest_call_without_sigz,TestDispatcherPickling.test_call_without_sig:sM7 !VL #zR %Urcddlm}|j|j|dd|j|j|tt t fy)Nr ) add_nopythonr8r9)r4r?rr-robject)rr?s rtest_call_nopythonz)TestDispatcherPickling.test_call_nopythonAs@4 sJO {VXW]W_L`arcTddlm}|j|j|tdy)Nr )add_nopython_fail)r )r4rCrr-r)rrCs rtest_call_nopython_failz.TestDispatcherPickling.test_call_nopython_failGs9 1BKQWXrcLddlm}|j|j|ddy)Nr )get_global_objmodeg@)g@)r4rGrr-)rrGs rtest_call_objmode_with_globalz4TestDispatcherPickling.test_call_objmode_with_globalLs: 1CS&Qrc\ddlm}|d}|j|j|ddy)Nr closure)rD)r4rKrr-)rrKinners rtest_call_closurez(TestDispatcherPickling.test_call_closurePs&/  6Brc \ddlm}|di|}|j|j|ddy)Nr )closure_with_globalsg@@)@)r4rQrr-)rjit_argsrQrNs rcheck_call_closure_with_globalsz6TestDispatcherPickling.check_call_closure_with_globalsUs*<$5H5 VDrc(|jdy)NT)nopythonrVrs r'test_call_closure_with_globals_nopythonz>TestDispatcherPickling.test_call_closure_with_globals_nopythonZ ,,d,;rc(|jdy)NT)forceobjrYrZs r&test_call_closure_with_globals_objmodez=TestDispatcherPickling.test_call_closure_with_globals_objmode]r\rc\ddlm}|d}|j|j|ddy)Nr )closure_calling_other_functionrTg&@)rSg@)r4rarr-)rrarNs r(test_call_closure_calling_other_functionz?TestDispatcherPickling.test_call_closure_calling_other_function`s'F.s3 jIrc\ddlm}|d}|j|j|ddy)Nr closure_calling_other_closurerTg @rR)r4rerr-)rrerNs r'test_call_closure_calling_other_closurez>TestDispatcherPickling.test_call_closure_calling_other_closurees'E-c2 VDrcLddlm}|j|j|ddy)Nr )dyn_func$rL)r4rhrr-)rrhs rtest_call_dyn_funcz)TestDispatcherPickling.test_call_dyn_funcjs0 2tDrcLddlm}|j|j|ddy)Nr )dyn_func_objmoderirj)r4rmrr-)rrms rtest_call_dyn_func_objmodez1TestDispatcherPickling.test_call_dyn_func_objmodeos8 1A2tLrc\ddlm}|d}|j|j||dy)Nr )get_renamed_module)rq)r4rprr-)rrpexpecteds rtest_renamed_modulez*TestDispatcherPickling.test_renamed_modulets-:&c* 1C (& 2rcddlm}|d}tj|}djdit }t jtjd|gy)z} Check that reconstructing doesn't depend on resources already instantiated in the original process. r rdrTzif 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_ F,S1,,t$# # s~~tT:;rcrddlm}|d}tj|}|d}tj|}tj|}tj|}tj|}|j |||j |ddd|j|j |ddd |j|||j |ddd |d }|d d tj|}~tjtj|}tj|}|j |||j |ddd |j|j |dddy)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[ 0qz,,t$ <<& LL ! LL ! LL " a 1a"%  1a"% A 1a"%qz R ,,t$  LL ! LL ! a 1a"%  1a"%rcTd}tjtjd|gy)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 s) 3 3 #N Vb Y RC E <<J E E M 2<$$&L.KlassNrrrclassvarrvrrKlassrHrrcd_yNd)rrsrmutatorzKTestCloudPickleIssues.test_dynamic_class_reset_on_unpickle..mutators  ENrc>jjdyr)rr)rrsrcheckzITestCloudPickleIssues.test_dynamic_class_reset_on_unpickle..checks   U^^S 1rr)rrrsavedrs` @r$test_dynamic_class_reset_on_unpicklez:TestCloudPickleIssues.test_dynamic_class_reset_on_unpicklesD   ! 2e    e   e  r__main__z*Test cannot run as when module is __main__ctd}|jt}|j|j d|j |j dy)Nspawn)target<timeoutr)rProcess"check_main_class_reset_on_unpicklestartjoinrexitcode)rmpprocs r!test_main_class_reset_on_unpicklez7TestCloudPickleIssues.test_main_class_reset_on_unpicklesJ !zz!CzD  "  *rcGdd}t|}td}|jt|f}|j |j d|j |jdy)NceZdZdZy)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_procsc  e  !zz!B%zR  "  *rcdtjt}|j|dy)N)rr)numbanjit issue_7356r)rcfuncs rtest_dynamic_class_issue_7356z3TestCloudPickleIssues.test_dynamic_class_issue_7356 s" :& *-rN) rrr__doc___numba_parallel_test_runittestskipIfrrrrvrrrrsG!(X__X+AC+C+ +.rrceZdZdZy)DynClassN)rrrr;rvrrrrs Arrctjd5dt_tj}dddtjd5tj}ddd|fS#1swY>S  %LEENUE"rcbt|}|jdk7sJd|_t||yr)r rr)rrs rrr0s. %LE >>S  ENUE"rcfd}|t|}|ur td|t|}|ur td|y)Nc:jdk7r tdy)NrzCheck failed. Klass reset.)rAssertionErrorrsrrz_check_dyn_class..check:s >>S  !=> > !rzExpected reuse)r r)rrrloadeds` rrr9sS? G 5\F U-.. G 5\F U-.. Grr)rrrrryrzrmultiprocessingrrnumba.core.errorsrnumba.tests.supportrnumba.core.target_extensionrnumba.cloudpicklerr r rrr@rrrrrrmainrvrrrs  ' )(C*v