wi;ddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z ddl Z ddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlmZddlZ ddlZddl m!Z"n#e#e$f$rdZdZ"YnwxYw ddl%Z%n #e#$rdZ%YnwxYwddl&m'Z'ddl(m)Z)ddl&m*Z*ddl&m+Z+ddl&m,Z,ddl-m.Z.dd l-m/Z/m0Z0dd l-m1Z1m2Z2dd l-m3Z3d d l4m5Z5d dl4m6Z6d dl4m7Z7d dl4m8Z8da9dZ:GddZ;e'j<fdZ=dZ>dZ?dZ@GddejAZBGddeBZCdZDdZEdZFd ZGd!ZHd"ZIeJd#krejKdSdS)$N) itemgetter attrgetterwrapspickle)register_pickle_by_value)unregister_pickle_by_value)list_registry_pickle_by_value)_should_pickle_by_reference)_make_empty_cellcell_set)_extract_class_dict _whichmodule)_lookup_module_and_qualname)subprocess_pickle_echo)subprocess_pickle_string)assert_run_python_script)subprocess_worker default_value another_valueceZdZdZdZdS)RaiserOnPicklec||_dSNexc)selfrs x/root/.openclaw/workspace/chatterbox_venv_py311/lib/python3.11/site-packages/srsly/tests/cloudpickle/cloudpickle_test.py__init__zRaiserOnPickle.__init__Ds c|jrrrs r __reduce__zRaiserOnPickle.__reduce__Gs hr"N)__name__ __module__ __qualname__r!r%r"r rrBs2r"rcRtjtj||S)zeHelper function to test whether object pickled with cloudpickle can be depickled with pickle protocol)rloads cloudpickledumps)objr,s r pickle_depickler1Ks$ < )#AAA B BBr"c.|ddS)N\z\\\\)replace) raw_filepaths r _escaper6Rs   g . ..r"cT ||n#t$rYnwxYw|Sr)remove ValueError)list_items r _maybe_remover<WsA  T      Ls  %%cvGddt}Gdd}Gdd||}t|}t|gdksJ|dd ksJ|d J|d ||ksJdS) NceZdZdZdZdS)"test_extract_class_dict..Az A docstringcdS)Nar)r$s r methodz)test_extract_class_dict..A.methodb3r"N)r&r'r(__doc__rBr)r"r Ar?`s)     r"rEceZdZdZdZdZdS)"test_extract_class_dict..Bz B docstring*cdS)Nbr)r$s r rBz)test_extract_class_dict..B.methodirCr"N)r&r'r(rD B_CONSTANTrBr)r"r BrGes.      r"rLceZdZdZdZdS)"test_extract_class_dict..C+cdS)Ncr)r$s r method_cz+test_extract_class_dict..C.method_corCr"N)r&r'r( C_CONSTANTrRr)r"r CrNls(      r"rT)rSrDrRrSrOrDrR)intrsortedkeysrR)rErLrTclsdicts r test_extract_class_dictrY_sC Aq "!$$G ',,.. ! !%J%J%J J J J J < B & & & & 9  % % % 7: qqss # #qqss||~~ 5 5 5 5 5 5r"ceZdZejZdZdZej e j dkpe jdko e jdkddZd Zd Zej d d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&d Z'd!Z(ej e j d"kd#d$Z)d%Z*d&Z+d'Z,d(Z-d)Z.d*Z/d+Z0d,Z1d-Z2d.Z3d/Z4d0Z5d1Z6d2Z7d3Z8d4Z9d5Z:d6Z;d7Zd:Z?ej e j@d;kre jdkpCe j d"kp0e jdko e jd?kpe jd.MyClassNr&r'r(r)r"r MyClassrn Dr"rpz)'del'-ed my_object has not been collected)weakrefrefr.r/)rrp my_objectwrs r %test_reducer_override_reference_cyclez5CloudPickleTest.test_reducer_override_reference_cyclest        GII [ # #)$$$ rtt||H|||||r"cZtd}td}t||j}|||||tdd}t||j}|||||dS)N rr+rrh)rangerr1r, assertEqual)rdgettergetter2s r test_itemgetterzCloudPickleTest.test_itemgetters "IIA!&4=AAA GGAJJ///Aq!!!&4=AAA GGAJJ/////r"cGdd}|}td}t||j}|||||tdd}t||j}|||||||_td}t||j}|||||tdd}t||j}|||||dS)NceZdZdZdS)*CloudPickleTest.test_attrgetter..Cc|Srr))rr;s r __getattr__z6CloudPickleTest.test_attrgetter..C.__getattr__s r"Nr&r'r(rr)r"r rTrs#     r"rTrAr+rJze.aze.b)rr1r,rze)rrTr{r|r}s r test_attrgetterzCloudPickleTest.test_attrgettersh         ACCC!&4=AAA GGAJJ///C%%!&4=AAA GGAJJ///accE""!&4=AAA GGAJJ///E5))!&4=AAA GGAJJ/////r"zRequires pytest -s to passctj}tjt j||j}|||dSNr+)sysstderrrr-r.r/r,rz)rout1out2s r test_pickling_file_handlesz*CloudPickleTest.test_pickling_file_handlessFz|K-dT]KKKLL t$$$$$r"cGdd}|atfdd}d|jjvt j|dS)NceZdZdZdS)6CloudPickleTest.test_func_globals..Unpicklablec td)Nz not picklable Exceptionr$s r r%zACloudPickleTest.test_func_globals..Unpicklable.__reduce__s000r"Nr&r'r(r%r)r"r Unpicklablers# 1 1 1 1 1r"rcDtjtjSr)r.r/exitr,r$sr z3CloudPickleTest.test_func_globals..s [-> 4=.*.*.*r"c.tjddS)Nr)rrr)r"r fooz.CloudPickleTest.test_func_globals..foos HQKKKKKr"r)r assertRaisesr assertTrue__code__co_namesr.r/)rrrs` r test_func_globalsz!CloudPickleTest.test_func_globalss 1 1 1 1 1 1 1 1 {}} )&*&*&*&* + + +    #,"77888#r"cN td}t||j}||t |tddd}t||j}||t |dS#t $rYdSwxYw)NHellor+rjrh)bufferr1r,rzstr NameError)r buffer_obj buffer_clones r test_bufferzCloudPickleTest.test_buffers J*: NNNL   \3z?? ; ; ;A..J*: NNNL   \3z?? ; ; ; ; ;    DD sBB B$#B$ctd}|t||j|dS)NsHellor+ memoryviewrzr1r,tobytesrrs r test_memoryviewzCloudPickleTest.test_memoryviewsQ))  dmLLL#++-- / / / / /r"cddd}t|}|||t|tjsJdSNrrjrArJ)rWr1rz isinstance_collections_abc dict_keysrrWresultss r test_dict_keyszCloudPickleTest.test_dict_keyss^Q$$&&!$'' $''''#3#=>>>>>>>r"cddd}t|}|t|t|t |t jsJdSr)valuesr1rzrVrr dict_valuesrrrs r test_dict_valuesz CloudPickleTest.test_dict_valuessjq!!((**!&)) &..999'#3#?@@@@@@@r"cddd}t|}|||t|tjsJdSr)itemsr1rzrr dict_itemsrrrs r test_dict_itemszCloudPickleTest.test_dict_itemss^a  &&((!%(( %((('#3#>???????r"ctjddg}t|}|||t |t |ksJdSN)rAr)rJrj) collections OrderedDictrWr1rztypers r test_odict_keyszCloudPickleTest.test_odict_keyssi&(';<<AACC!$'' $'''DzzT']]******r"ctjddg}t|}|t |t |t |t |ksJdSr)rrrr1rzlistrrs r test_odict_valuesz!CloudPickleTest.test_odict_valuessu((H)=>>EEGG!&)) gV 555F||tG}},,,,,,r"ctjddg}t|}|||t |t |ksJdSr)rrrr1rzrrs r test_odict_itemsz CloudPickleTest.test_odict_itemssi'8(<==CCEE!%(( %(((E{{d7mm++++++r"ctdddd}|t||j|dS)NsHello!Hello!Hello!rjr+rrs r )test_sliced_and_non_contiguous_memoryviewz9CloudPickleTest.test_sliced_and_non_contiguous_memoryviews\ ..qAv6  dmLLL#++-- / / / / /r"ctdtdz}|t||j|dS)NsHello!gcAr+)rrUrzr1r,rrs r test_large_memoryviewz%CloudPickleTest.test_large_memoryview s\ CHH 455  dmLLL#++-- / / / / /r"cl|td|jddS)NcdSNrr)r)r"r rz-CloudPickleTest.test_lambda..sr"r+rrzr1r,r$s r test_lambdazCloudPickleTest.test_lambdasG B DMBBBDDa I I I I Ir"cd\fdfd}|t||jdddS)N)rrjc|zSrr))xrAs r rz5CloudPickleTest.test_nested_lambdas..s q1ur"c |zSrr))rrJf1s r rz5CloudPickleTest.test_nested_lambdas..srr!uuzr"r+rr)rf2rArJrs @@@r test_nested_lambdasz#CloudPickleTest.test_nested_lambdass`1 ___ ! ! ! ! ! DdmDDDQGGKKKKKr"cd}d}t||j}|||t|d|j}||dddS)NcfdS)NcSrr)gsr rz=CloudPickleTest.test_recursive_closure..f1..gsr"r)rs@r rz2CloudPickleTest.test_recursive_closure..f1s!     Hr"cfdS)Nc6|dkrn||dz zSrr))nbasers r rz=CloudPickleTest.test_recursive_closure..f2..gs% Avvtt1qqQxx<7r"r))rrs`@r rz2CloudPickleTest.test_recursive_closure..f2s' 8 8 8 8 8 8Hr"r+rjr1r,rz)rrrg1g2s r test_recursive_closurez&CloudPickleTest.test_recursive_closures        RRTTDM : : : r""" RRUUT] ; ; ; A$$$$$r"cd}||jdudt||j}||jduddS)NcdS)z-a function with no closure cells Nr)r)r"r fz9CloudPickleTest.test_closure_none_is_preserved..f*r"zf actually has closure cells!msgr+z.g now has closure cells even though f does not)r __closure__r1r,rrrs r test_closure_none_is_preservedz.CloudPickleTest.test_closure_none_is_preserved)s     MT !/    A 6 6 6  MT !@      r"c:d}|}tjt5|dddn #1swxYwYt||j}tjt5|ddddS#1swxYwYdS)Nc fd}|S)NFc dSrr))cellsr rz?CloudPickleTest.test_empty_cell_preserved..f..g?s r"r))rrs @r rz4CloudPickleTest.test_empty_cell_preserved..f;s(      Hr"r+)pytestraisesrr1r,)rrrrs r test_empty_cell_preservedz)CloudPickleTest.test_empty_cell_preserved:s#   QSS ]9 % %   BDDD               R$- 8 8 8 ]9 % %   BDDD                  s! >AA8 BBBcd}t||j}||ddS)Ncddhfd}|S)Nrrjc"tSr)len)ssr rz=CloudPickleTest.test_unhashable_closure..f..gPs1vv r"r))rrs @r rz2CloudPickleTest.test_unhashable_closure..fMs,AA     Hr"r+rjrrs r test_unhashable_closurez'CloudPickleTest.test_unhashable_closureLsO    AACC$- 8 8 8 a     r"cGdd}Gdd|}||dt||j}||d||jd|}t||j}||ddS)NceZdZdZdS)NCloudPickleTest.test_dynamically_generated_class_that_uses_super..BasecdSrr)r$s r rBzUCloudPickleTest.test_dynamically_generated_class_that_uses_super..Base.method[sqr"Nr&r'r(rBr)r"r BaserZs#     r"rc"eZdZdZfdZxZS)QCloudPickleTest.test_dynamically_generated_class_that_uses_super..DerivedDerived DocstringcJtdzSr)superrB)r __class__s r rBzXCloudPickleTest.test_dynamically_generated_class_that_uses_super..Derived.method`sww~~''!++r")r&r'r(rDrB __classcell__)r s@r Derivedr ^s>   , , , , , , , , ,r"rrjr+r )rzrBr1r,rD)rrrUnpickledDerivedorig_dr{s r 0test_dynamically_generated_class_that_uses_superz@CloudPickleTest.test_dynamically_generated_class_that_uses_superXs#         , , , , ,d , , , ))++Q///+7T]KKK ))++2244a888 )13FGGG FT] ; ; ; Q'''''r"c$Gdd}||_||_t||j}t|}||d||d||jd||ddS)NceZdZdZdS):CloudPickleTest.test_cycle_in_classdict_globals..CcdS)Nwoohoo!r)r$s r it_workszCCloudPickleTest.test_cycle_in_classdict_globals..C.it_worksvs yr"N)r&r'r(rr)r"r rTrts# ! ! ! ! !r"rTr+r)C_again instance_of_Cr1r,rzr)rrT depickled_Cdepickled_instances r test_cycle_in_classdict_globalsz/CloudPickleTest.test_cycle_in_classdict_globalsrs ! ! ! ! ! ! ! !  !##%a$-@@@ ,QQSS11 //119=== ,,..77999EEE 2;;==yIII +4466 BBBBBr"cdfd|t|jddd|t|jdddfd Gfd d }t||j}||dd||d dd t ||j}||d dd |t|dd|t|d dd t |d |j}||dd |t|djd|t|d jdd t |d j|j}||dd dS)NrHcLttd||zz S)Nrx)sumry)ryLOCAL_CONSTANTs r some_functionzNCloudPickleTest.test_locally_defined_function_and_class..some_functions# b NNNE^+ +r"r+)rQrhrjcSrr))r"sr rzICloudPickleTest.test_locally_defined_function_and_class...r"c0eZdZdZdZfdZfdZdS)JCloudPickleTest.test_locally_defined_function_and_class..SomeClassz:Overly complicated class with nested references to symbolsc||_dSr)valuerr+s r r!zSCloudPickleTest.test_locally_defined_function_and_class..SomeClass.__init__s " r"cz Srr))rr"hidden_constants r onezNCloudPickleTest.test_locally_defined_function_and_class..SomeClass.ones%(9(999r"cV||dz|jzSr)r/r+)rrr#s r some_methodzVCloudPickleTest.test_locally_defined_function_and_class..SomeClass.some_methods)xxzzMM!Q$7$77$*DDr"N)r&r'r(rDr!r/r1)r"r.r#sr SomeClassr)sh L L # # # : : : : : : E E E E E E Er"r2r)rzr1r,r/r1r)rr2 clone_class new_instance new_methodr"r.r#s @@@r 'test_locally_defined_function_and_classz7CloudPickleTest.test_locally_defined_function_and_classs , , , , , OOOOPRTUVVXYZZZ OOOOPRTUVVXYZZZ0000 E E E E E E E E E E E E&i$-HHH  Q++--q111 Q33B77;;;,YOOO  Q33B77;;; 166::<.some_generators03ZZ    r"r+rh)r1r,rrry)rrWgen2s r test_generatorzCloudPickleTest.test_generators   ~ FFFDDGG}}^^A%6%6 7 77777DDGG}}U1XX......r"c`Gdd}|jd}|jd}t||j|_t||j|_||d||ddS)Nc:eZdZedZedZdS)+CloudPickleTest.test_classmethod..AcdS)Nsmr)r)r"r test_smz3CloudPickleTest.test_classmethod..A.test_smtr"cdSNcmr)clss r test_cmz3CloudPickleTest.test_classmethod..A.test_cmr`r"N)r&r'r( staticmethodr_ classmethodrfr)r"r rEr\sF   \    [   r"rEr_rfr+r^rc)__dict__r1r,r_rfrz)rrEr^rcs r test_classmethodz CloudPickleTest.test_classmethods        Z " Z "#B??? #B???  d+++ d+++++r"cGdd}t|j|j|_||ddS)Nc$eZdZedZdS)1CloudPickleTest.test_bound_classmethod..AcdSrbr)rds r rfz9CloudPickleTest.test_bound_classmethod..A.test_cmr`r"N)r&r'r(rhrfr)r"r rErms-   [   r"rEr+rc)r1rfr,rz)rrEs r test_bound_classmethodz&CloudPickleTest.test_bound_classmethodsh         $AI FFF  d+++++r"cvttj}||dddS)NabcABC)r1rupperrz)rrs r test_method_descriptorsz'CloudPickleTest.test_method_descriptorss5 CI & & 55)))))r"cGdd}t|j|j}||j|jjdS)NceZdZdZdS).Fc |dzSrr)rrs r rz>CloudPickleTest.test_instancemethods_without_self..F.fs 1u r"N)r&r'r(rr)r"r Frws#     r"rzr+)r1rr,rzr&)rrzrs r !test_instancemethods_without_selfz1CloudPickleTest.test_instancemethods_without_selfsd         AC$- 8 8 8 QS\22222r"crtt|j}|t|dSr)r1rr,rz)r pickle_clones r test_modulezCloudPickleTest.test_module s2&v FFF  .....r"ctjd}d}ttj||jt ||j}||j |j || d| d|| d| dtjdkrt||j}||j |j || d| d|| d| dt ||g\}}|t!|t!| |t"jd<t |j |j}|| d|dt"jdddS#t"jddwxYw)Nmodz x = 1 def f(y): return x + y class Foo: def method(self, x): return f(x) r+rr@)types ModuleTypeexectextwrapdedentrir1r,rzrrFoorBplatformpython_implementationridrmodulespop)rrcodemod2mod3mod1 depickled_fs r test_dynamic_modulez#CloudPickleTest.test_dynamic_modules?u%%  X_T " "CL111sT];;; ''' q466!99--- ))!,,dhhjj.?.?.B.BCCC  ) + +v 5 5)# FFFD   SUDF + + +   SUU1XXtvvayy 1 1 1   SWWYY--a00$((**2C2CA2F2F G G G%c3Z00 d D2d88,,, )!$CK )#%$-HHHK   SUU1XX{{1~~ 6 6 6 KOOE4 ( ( ( ( (CKOOE4 ( ( ( (s >AJ<<"Kctj|jd}d}|t |} ddlm}|}t|d5}tj |||j dddn #1swxYwYttj|tj|dS#tj|wxYw)Nzlocal_func_g.pklz from srsly.cloudpickle.compat import pickle import gc with open("{pickled_func_path}", 'rb') as f: func = pickle.load(f) assert func(range(10)) == 45 )pickled_func_pathr)make_local_functionwbr+)ospathjoinraformatr6!srsly.tests.cloudpickle.testutilsropenr.rNr,rrrunlink)rrchild_process_scriptrrrs r test_module_locals_behaviorz+CloudPickleTest.test_module_locals_behavior6s; GLL6HII  4::")*;"<"< ; > > ) M M M M M M##%%A'.. ?! A >>>> ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? %X_5I%J%J K K K I' ( ( ( ( (BI' ( ( ( (s0 C,B CBCB$CC-cGdd}tjd}td|j|ddksJd|jvsJ|}t jt5tj |dddn #1swxYwYt|jdtr||jdd<n7t|jdtjr||jd_ t||j }d|jvsJt|jdtr d |jvsJn.UnpickleableObjectc td)NzUnpicklable object)r9r$s r r%zcCloudPickleTest.test_dynamic_module_with_unpicklable_builtin..UnpickleableObject.__reduce__]s !5666r"Nrr)r"r UnpickleableObjectr\s# 7 7 7 7 7r"rrzf = lambda x: abs(x)r __builtins__unpickleable_objr+abs)rrrrirrrr9r.r/rdictrr1r,rhasattr)rrrunpicklable_obj depickled_mods r ,test_dynamic_module_with_unpicklable_builtinz2D 9 9 L?NCL (); < <  ^4e6F G G LM666666 &~68HJJ >=5u== = ==r""a'''' uuRyyA~~~~~~sB!!B%(B%cFtjd}d}ttj||jt j|j d}t j|j d}d}d}| |}| t|t|t||j } t|d 5}tj|||j dddn #1swxYwYt!tj|t j|rt j|t j|rt j|dSdS#t j|rt j|t j|rt j|wwxYw) Nrz x = 1 z&dynamic_module_from_parent_process.pklz%dynamic_module_from_child_process.pklau from srsly.cloudpickle.compat import pickle import textwrap import srsly.cloudpickle as cloudpickle from srsly.tests.cloudpickle.testutils import assert_run_python_script child_of_child_process_script = {child_of_child_process_script} with open('{parent_process_module_file}', 'rb') as f: mod = pickle.load(f) with open('{child_process_module_file}', 'wb') as f: cloudpickle.dump(mod, f, protocol={protocol}) assert_run_python_script(textwrap.dedent(child_of_child_process_script)) z ''' from srsly.cloudpickle.compat import pickle with open('{child_process_module_file}','rb') as fid: mod = pickle.load(fid) ''' )child_process_module_file)parent_process_module_filerchild_of_child_process_scriptr,rr+)rrrrrrirrrrarr6r,rr.rNrexistsr)rrrrrrrfids r .test_load_dynamic_module_in_grandchild_processz>CloudPickleTest.test_load_dynamic_module_in_grandchild_processsb u%%  X_T " "CL111 &(W\\ KA&C&C"$&GLL K@%B%B! ()%)F(L(L*C)M)E)E% 4::'./I'J'J&-.G&H&H*12O*P*P] ; $ $ 50$77 C3 cDMBBBB C C C C C C C C C C C C C C C %X_5I%J%J K K Kw~~899 6 4555w~~788 5 344444 5 5w~~899 6 4555w~~788 5 34444 5s1(F78D! F7!D%%F7(D%)$F77A)H cdd}fd}tj||j}d|vsJd|vsJd|vsJd|vsJdS) Nc |dzSrr)rs r nested_functionzDCloudPickleTest.test_correct_globals_import..nested_function q5Lr"c*tj|Sr)mathexprs r unwanted_functionzFCloudPickleTest.test_correct_globals_import..unwanted_functions8A;; r"c ||zSrr))rr!rs r my_small_functionzFCloudPickleTest.test_correct_globals_import..my_small_functions"?1%%) )r"r+smy_small_functionsnested_functionsunwanted_functionsmath)r.r/r,)rrrrJrs @r test_correct_globals_importz+CloudPickleTest.test_correct_globals_imports       * * * * *  /$- H H H$q((((!Q&&&&#1,,,,ar"ctjdddlm}ddl}ddl}ddl}t|sJt|jsJt|sJt|j sJtj d}t|rJtj dkrddl}t|sJddlmcm}t|sJt%||j|usJddlm}t|sJt%||j|usJtjt.5ddl}dddn #1swxYwYddlm} t| rJt5| d rJt%| |j} | | usJ| d | d ksJddlmcmcm} t| sJt%| |j| usJd } tjt<| 5tt?ddddS#1swxYwYdS) N_cloudpickle_testpkgrrdynamic_moduler@r+)dynamic_submodule_two)submodule_threer'rz.cannot check importability of object instancesmatch) r importorskipsrsly.cloudpickle.compatros.pathrcollections.abcr rrqrrrr_codecs*_cloudpickle_testpkg.mod.dynamic_submodulerdynamic_submoduler1r,_cloudpickle_testpkg.modrr ImportError(_cloudpickle_testpkg.mod.submodule_threerrr?_cloudpickle_testpkg.mod.dynamic_submodule.dynamic_subsubmoduledynamic_subsubmodule TypeErrorobject) rrrrrrmm2rm3 depickled_m3r^expecteds r test_module_importabilityz)CloudPickleTest.test_module_importabilitysu2333333333*622222*2733333*;77777*;?;;;;;)*:;;.~>>>>>  ) + +v 5 5 NNN.w77 7 77 ?>>>>>>>>*1-----q4=999Q>>>> IHHHHH+2.....rDM:::b@@@@]; ' ' < < ; ; ; ; < < < < < < < < < < < < < < <BBBBBB.r222222|,,,,,&rDMBBB 2%%%%ttAww,..++++++ UTTTTTTTTTTT*2.....rDM:::b@@@@C ]9H 5 5 5 2 2 ' 1 1 1 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2s$EEE'IIIcn|ttt|jdSr)rzEllipsisr1r,r$s r test_EllipsiszCloudPickleTest.test_Ellipsiss= (DMJJJ L L L L Lr"crtt|j}|t|dSr)r1NotImplementedr,rz)rExcClones r test_NotImplementedz#CloudPickleTest.test_NotImplemented!s2">DMJJJ 22222r"cttd|j}|td|dSr)r1rr,rzrress r test_NoneTypezCloudPickleTest.test_NoneType%s>d4jj4=AAA dS)))))r"cttt|j}|tt|dSr)r1rrr,rzrs r test_EllipsisTypez!CloudPickleTest.test_EllipsisType)s>d8nnt}EEE h-----r"cttt|j}|tt|dSr)r1rrr,rzrs r test_NotImplementedTypez'CloudPickleTest.test_NotImplementedType-sBd>22T]KKK n--s33333r"ctt|jtusJddlm}t||j|usJdS)Nr+r)mkdir)r1zipr,rr)rrs r test_builtin_functionz%CloudPickleTest.test_builtin_function1s\ sT];;;sBBBBut}===FFFFFFr"ctttttt fD]8}t |j|j}t|||sJ9dSr) rtupleset frozensetrrr1__new__r,r)rt cloned_news r test_builtin_type_constructorz-CloudPickleTest.test_builtin_type_constructor<sa uc9dF: 0 0A(T]KKKJjjmmQ// / // / 0 0r"cLd}|j}t|j}t|jd}||usJt||j}t||j}t||j}||ksJ||usJ||usJdS)N?hexr+)rrrir1r,)rr0bound_classicmethodunbound_classicmethodclsdict_classicmethoddepickled_bound_methdepickled_unbound_methdepickled_clsdict_meths r test_builtin_classicmethodz*CloudPickleTest.test_builtin_classicmethodQs!g $S  $S 25 9$(=====. $- 9 9 9!0 !DM";";";!0 !DM";";"; $#%%)<)<)>)>>>>>%)>>>>>%)>>>>>>>r"aarch64Nrj)rhrx)rhrxri)rh zFails on aarch64 + python 3.10+ in cibuildwheel, currently unable to replicate failure elsewhere; fails sometimes for pypy on conda-forge; fails for python 3.10.8+ and 3.11+cd}|j}t|j}t|jd}t||j}t||j}t||j}d}||||ksJ||||ksJt jdkr |||t|ksJt jdkr\t|t|ksJ|t||t|ksJdSdS)Nrfromhexr+0x1rgr@) rrrir1r,rrfloat__func__) rr0bound_clsmethodunbound_clsmethodclsdict_clsmethodrrrargs r test_builtin_classmethodz(CloudPickleTest.test_builtin_classmethodhs+ II- )% / dm 5 5 5!0  "7"7"7!0  "7"7"7 $#C((OOC,@,@@@@@%%c**.?.?.D.DDDDD  ) + +y 8 8*)#..2C2CE32O2OOOOO  ) + +v 5 5 .//48I3J3JJJJJ)22s099%EEFFFF 6 5FFr"c@d}|j}t|j}t|jd}t||j}t||j}t||j}||ksJ||usJ||usJdS)Nr__repr__r+)r rrir1r,)rr0bound_slotmethodunbound_slotmethodclsdict_slotmethodrrrs r test_builtin_slotmethodz'CloudPickleTest.test_builtin_slotmethods<!#YY/!#YY/ ;. t} 6 6 6!0 "8"8"8!0 "8"8"8 $#%%)9)9););;;;;%);;;;;%);;;;;;;r"z0No known staticmethod example in the pypy stdlibcxd}|j}t|j}t|jd}||usJt||j}t||j}t||j}||usJ||usJ|j|jusJt|t|udS)Nr maketransr+)rrrir1r,r)rr0bound_staticmethodunbound_staticmethodclsdict_staticmethodrrrs r test_builtin_staticmethodz)CloudPickleTest.test_builtin_staticmethods ]#Cyy2#Cyy1+>!%99999.  8 8 8!0 4=":":":!0 4=":":":$'99999%)=====&.2F2OOOOO #$$-A(B(BBBBBr"ztest needs Tornado installedc ddlm m} j fd j fd}t j||g|j}dx}tj|\}}| ||u|j }| tj|d}||ddS)Nr)genioloopc3fK|V|dzr)sleepReturn)rr!rs r rz1CloudPickleTest.test_tornado_coroutine..fs3))A,,   **QU## #r"c3TKd|V}|dz)Ng{Gz?r)r)r!rrrs r rz1CloudPickleTest.test_tornado_coroutine..gs2$ ""C**S1W%% %r"r+rr3)tornadorr coroutiner.r/r,rr-rIOLoopcurrentrun_syncr:r;rz) rrrdatarg3looprrrs @@r test_tornado_coroutinez&CloudPickleTest.test_tornado_coroutines  ('''''''  $ $ $ $  $  & & & & &  & !Q$-@@@ Ad##B b!!!}$$&&mmI-b!4455 a     r")rhrrbeta)rhrrr(r9z.https://github.com/python/cpython/issues/92932c d}dtd|D}tjd fd|D}dd|}t t j||||d }|}tj ||g|j }dx}}tj |\}} | || u|||dS) Nicg|]}d|zS)zg%dr)).0rVs r z5CloudPickleTest.test_extended_arg..s444q444r"rrHc<i|]}|dS)d) randrange)r+namers r z5CloudPickleTest.test_extended_arg..s' 6 6 6T1;;s## 6 6 6r"zx import zlib def f(): x = {tup} return zlib.crc32(bytes(bytearray(x))) z, )tuprr+)ryrandomRandomrrrrrr.r/r,rr-rrz) rnvarsnamesr{rrrr$rf3r1s @r test_extended_argz!CloudPickleTest.test_extended_args44E!UOO444 M"   6 6 6 6 6 6 6 Ftyy''F ( (  X_T " "Aq))) cFacc !Q$-@@@ Ad##B b!!! s#####r"cddlad}tj||j}bt t jD].}|dddkr t j|=/tj |}|dS)Nrc2tjjj}dSrxmletree ElementTreeCommentrs r examplez/CloudPickleTest.test_submodule..examples %-AAAr"r+.r=) xml.etree.ElementTreer=r.r/r,rrrsplitrr-)rrArr;rs r test_submodulezCloudPickleTest.test_submodules %$$$ . . .  g > > > %% & &Dzz#q!U**K% LOO r"c"d}|}tj||j}ttjD].}|dddkr tj|=/tj|}|dS)Ncddlfd}|S)Nrc*jjj}dSr)r>r?r@)rr=s r rAzFCloudPickleTest.test_submodule_closure..scope..examplesI)1r")rC)rAr=s @r scopez5CloudPickleTest.test_submodule_closure..scopes- ( ( ( ( 2 2 2 2 2Nr"r+rBrr=)r.r/r,rrrrDr-)rrIrArr;rs r test_submodule_closurez&CloudPickleTest.test_submodule_closures    %''  g > > >%% & &Dzz#q!U**K%  a  r"cd}ddla|}tj||j}dt j|dzdz}tj tj d|grJdS)Nc d}|S)Nc2tjjj}dSrr<rs r rAzACloudPickleTest.test_multiprocess..scope..example(sI)1r"r))rAs r rIz0CloudPickleTest.test_multiprocess..scope's 2 2 2Nr"rr+[import base64; from srsly.cloudpickle.compat import pickle; pickle.loads(base64.b32decode('ascii'))()-c) rCr=r.r/r,base64 b32encodedecode subprocesscallr executablerrIrArcommands r test_multiprocessz!CloudPickleTest.test_multiprocess%s    %$$$%''  g > > >5#A&&--g667  ?CND'#BCCCCCCCr"cd}|}ddlmcmatj||j}dt j|dzdz}tj tj d|grJdS)Ncddlmfd}|S)Nrc.tj}j}dSr)r>r@r?)rfoobars r rAz;CloudPickleTest.test_import..scope..example@sM&r") xml.etreer>)rAr^s @r rIz*CloudPickleTest.test_import..scope>s5 & & & & & & ' ' ' ' 'Nr"rr+rNrOrPrQ) rCr>r?r.r/r,rRrSrTrUrVrrWrXs r test_importzCloudPickleTest.test_import:s    %''---------  g > > >5#A&&--g667  ?CND'#BCCCCCCCr"ctj}tjtd5t j|ddddS#1swxYwYdS)Nz4only be shared between processes through inheritancer)multiprocessingLockrr RuntimeErrorr.r/)rlocks r test_multiprocessing_lock_raisesz0CloudPickleTest.test_multiprocessing_lock_raisesPs#%% ].MyUnrelatedClassNror)r"r MyUnrelatedClassrrqr"rceZdZdS)BCloudPickleTest.test_abc_cache_not_pickled..MyRelatedClassNror)r"r MyRelatedClassrrqr"rsMyUnrelatedClasssMyRelatedClass)rqABCMetaregister issubclassr.r/r-)rrprrrdepickled_classs r test_abc_cache_not_pickledz*CloudPickleTest.test_abc_cache_not_pickleds+iR00                 (((.88888.'22222  g & &"!++++ A%%%%%+A...@@@@@./:::::::r"cGddtj}Gfdd||t}t ||j}t |j}t |}t t|sJt t|sJ||d||d|| d|| d|| d|| d||j d||j d| t|Gfdd||d Gd d |}| t|dS) NceZdZejdZeejdZeejdZ e ejdZ dS)/CloudPickleTest.test_abc..AbstractClasscdSzA methodNr)r$s r r1z;CloudPickleTest.test_abc..AbstractClass.some_methodrr"cdSz A classmethodNr)rds r some_classmethodz@CloudPickleTest.test_abc..AbstractClass.some_classmethodrr"cdSzA staticmethodNr)r)r"r some_staticmethodzACloudPickleTest.test_abc..AbstractClass.some_staticmethodrr"cdSz A propertyNr)r)r"r some_propertyz=CloudPickleTest.test_abc..AbstractClass.some_propertyrr"N) r&r'r(rqabstractmethodr1rhrrgrpropertyrr)r"r AbstractClassrs         $ $ [ $   % % \ %   ! ! X ! ! !r"rc\eZdZdZefdZedZedZ dS)/CloudPickleTest.test_abc..ConcreteClasscdSN it works!r)r$s r r1z;CloudPickleTest.test_abc..ConcreteClass.some_method"{r"c|ksJdSrr)re ConcreteClasss r rz@CloudPickleTest.test_abc..ConcreteClass.some_classmethodm++++"{r"cdSrr)r)r"r rzACloudPickleTest.test_abc..ConcreteClass.some_staticmethod"{r"cdSrr)r$s r rz=CloudPickleTest.test_abc..ConcreteClass.some_propertyrr"N r&r'r(r1rhrrgrrrrsr rr # # # # # # #[ # # #\ # # #X # # #r"rr+rc\eZdZdZefdZedZedZ dS)7CloudPickleTest.test_abc..DepickledBaseSubclasscdSNit works for realz!r)r$s r r1zCCloudPickleTest.test_abc..DepickledBaseSubclass.some_method,,r"c|ksJdSrr)reDepickledBaseSubclasss r rzHCloudPickleTest.test_abc..DepickledBaseSubclass.some_classmethod33333,,r"cdSrr)r)r"r rzICloudPickleTest.test_abc..DepickledBaseSubclass.some_staticmethod,,r"cdSrr)r)r"r rzECloudPickleTest.test_abc..DepickledBaseSubclass.some_propertyrr"Nrrsr rr - - - - - - -[ - - -\ - - -X - - -r"rrceZdZdZdS)8CloudPickleTest.test_abc..IncompleteBaseSubclasscdSNz&this class lacks some concrete methodsr)r$s r r1zDCloudPickleTest.test_abc..IncompleteBaseSubclass.some_method??r"Nr&r'r(r1r)r"r IncompleteBaseSubclassr( @ @ @ @ @r"rrqrrrrr1r,rrzr1rrrrr rrconcrete_instancedepickled_baserrrrrs @@r test_abczCloudPickleTest.test_abcs ! ! ! ! !CG ! ! !( # # # # # # #M # # #& u%%%)MOO(OOO)-37=BBB,->??%/////%00000 **6688+FFF +7799;GGG 99;;[III +<<>> LLL **<<>> LLL +==??MMM **8+FFF +9;GGG )^444 - - - - - - -N - - -" ..00<<>>. 0 0 0 @ @ @ @ @^ @ @ @ )%;<<<<.AbstractClasscdSrr)r$s r r1zACloudPickleTest.test_abstracts..AbstractClass.some_methodrr"cdSrr)rds r rzFCloudPickleTest.test_abstracts..AbstractClass.some_classmethodrr"cdSrr)r)r"r rzGCloudPickleTest.test_abstracts..AbstractClass.some_staticmethodrr"cdSrr)r$s r rzCCloudPickleTest.test_abstracts..AbstractClass.some_property rr"N) r&r'r(rqrr1abstractclassmethodrabstractstaticmethodrabstractpropertyrr)r"r rrs       $ $ $% $ $ % % %& % % ! ! !" ! ! ! !r"rc\eZdZdZefdZedZedZ dS)5CloudPickleTest.test_abstracts..ConcreteClasscdSrr)r$s r r1zACloudPickleTest.test_abstracts..ConcreteClass.some_methodrr"c|ksJdSrr)rs r rzFCloudPickleTest.test_abstracts..ConcreteClass.some_classmethodrr"cdSrr)r)r"r rzGCloudPickleTest.test_abstracts..ConcreteClass.some_staticmethodrr"cdSrr)r$s r rzCCloudPickleTest.test_abstracts..ConcreteClass.some_propertyrr"Nrrsr rr rr"rr+rc\eZdZdZefdZedZedZ dS)=CloudPickleTest.test_abstracts..DepickledBaseSubclasscdSrr)r$s r r1zICloudPickleTest.test_abstracts..DepickledBaseSubclass.some_method9rr"c|ksJdSrr)rs r rzNCloudPickleTest.test_abstracts..DepickledBaseSubclass.some_classmethod<rr"cdSrr)r)r"r rzOCloudPickleTest.test_abstracts..DepickledBaseSubclass.some_staticmethodArr"cdSrr)r$s r rzKCloudPickleTest.test_abstracts..DepickledBaseSubclass.some_propertyErr"Nrrsr rr8rr"rrceZdZdZdS)>CloudPickleTest.test_abstracts..IncompleteBaseSubclasscdSrr)r$s r r1zJCloudPickleTest.test_abstracts..IncompleteBaseSubclass.some_methodMrr"Nrr)r"r rrLrr"rrrs @@r test_abstractszCloudPickleTest.test_abstractss ! ! ! ! !CG ! ! !" # # # # # # #M # # #& u%%%)MOO(OOO)-37=BBB,->??%/////%00000 **6688+FFF +7799;GGG 99;;[III +<<>> LLL **<<>> LLL +==??MMM **8+FFF +9;GGG )^444 - - - - - - -N - - -" ..00<<>>. 0 0 0 @ @ @ @ @^ @ @ @ )%;<<<<.SomeClassc||_dSrrrys r r!zNCloudPickleTest.test_weakset_identity_preservation..SomeClass.__init__Xs r"Nr&r'r(r!r)r"r r2rWs#     r"r2rrjrhr+)rrWeakSetr1r,rzrrr) rr2obj1obj2obj3thingsresultweakset depickled1 depickled2 depickled3s r "test_weakset_identity_preservationz2CloudPickleTest.test_weakset_identity_preservationRs         %9Q<<1yy||Dd/4,//tTB $-@@@6<3Z q))) q))) q))) Wq))) W J'?@@@@@r"c$dd_Gfdd}|}ddksJ|jusJtjt5|j}dddn #1swxYwY |t jd<td}|dksJ|Jt|j }|ddksJ t j ddS#t j dwxYw)Nc |dzS)Nrjr)rs r funczMCloudPickleTest.test_non_module_object_passing_whichmodule_test..funcps 6Mr"c eZdZdZfdZdS)XCloudPickleTest.test_non_module_object_passing_whichmodule_test..NonModuleObjectcd|_dSr) some_attrr$s r __ini__z`CloudPickleTest.test_non_module_object_passing_whichmodule_test..NonModuleObject.__ini__xs !%r"c"|dkrSt)Nr)AttributeError)rr0rs r rzdCloudPickleTest.test_non_module_object_passing_whichmodule_test..NonModuleObject.__getattr__{s 6>>K((r"N)r&r'r(rr)rsr NonModuleObjectrws= & & & ) ) ) ) ) ) )r"rrjr9r+) r'rrrrrrrrr1r,r)rrnon_module_objectrzfunc_module_namedepickled_funcrs @r /test_non_module_object_passing_whichmodule_testz?CloudPickleTest.test_non_module_object_passing_whichmodule_testis     ) ) ) ) ) ) ) ) ) ),O--tAww!||||(-----]> * * , ,!+A , , , , , , , , , , , , , , , /->CK) *+D$77 #'88888#+++,TDMJJJN!>!$$))))) KOO- . . . . .CKOO- . . . .sA..A25A2:AC..!DcfttjfD]}dD]Gdd|}Gfdd}d}|_|tjur |d}n |}|tjd< |t|dt||j }||dtj dd#tj ddwxYwdS) N)_missing_moduleNceZdZdZdS)BCloudPickleTest.test_unrelated_faulty_module..FaultyModulectrrrr0s r rzNCloudPickleTest.test_unrelated_faulty_module..FaultyModule.__getattr__s(kk)r"Nrr)r"r FaultyModuler s#*****r"r ceZdZZdZdS)9CloudPickleTest.test_unrelated_faulty_module..FoocdSrr)r$s r rz=CloudPickleTest.test_unrelated_faulty_module..Foo.foos*{r"N)r&r'r(r) module_namesr rrs)!,J+++++r"rcdSrr)r)r"r rz9CloudPickleTest.test_unrelated_faulty_module..foos&;r"_faulty_modulerr+) rrrr'rrrzr1rr,r)r base_classr rr faulty_moduleclonedrs @r test_unrelated_faulty_modulez,CloudPickleTest.test_unrelated_faulty_modules "5#34! <! .fs% s##A5577Nr") CloudPickler)cloudpickle_fastr+r) rsrsly.cloudpicklerr_module_reducedispatch_tablerrr.r-r/r,rz)rrrcp_fastrrrs @r test_dynamic_pytest_modulez*CloudPickleTest.test_dynamic_pytest_modules        322222AAAAAA8?8N #D$4$45  k/DMJJJ K K F#####r"cxd}t||j}||j|jdS)Nc|Srr)rs r rz;CloudPickleTest.test_function_module_name..sr"r+)r1r,rzr'rrrs r test_function_module_namez)CloudPickleTest.test_function_module_names>{  >>> *DO<<<<.funcsHr"r(r+z)rr1r,rzr(r$s r test_function_qualnamez&CloudPickleTest.test_function_qualnames    4 ( ( E$TDMBBBF   V0$2C D D D/  >>> ,d.?@@@@@r"cGdd}|}|jdksJ|jjdksJtjt5d|_dddn #1swxYwYd|_t |}|jdksJ|jdksJtjt5d|_dddn #1swxYwYd|_|jdksJt|jjdkdS)NcbeZdZdZdZedZedZejdZdS)/CloudPickleTest.test_property..MyObjectrc|jS)A read-only attribute)_read_only_valuer$s r read_only_valuez?CloudPickleTest.test_property..MyObject.read_only_values ,,r"c|jSr_read_write_valuer$s r read_write_valuez@CloudPickleTest.test_property..MyObject.read_write_values --r"c||_dSrr1r,s r r3z@CloudPickleTest.test_property..MyObject.read_write_values).&&&r"N) r&r'r(r.r2rr/r3setterr)r"r MyObjectr+ss  !   - -X - . .X . $ / /% $ / / /r"r6rr-rjrh)r/rDrrrr3r1r)rr6rt depickled_objs r test_propertyzCloudPickleTest.test_propertys / / / / / / / /&HJJ (A----'/3JJJJJ ]> * * * *()I % * * * * * * * * * * * * * * *%& "' 22 ,1111-2222]> * * * *()I % * * * * * * * * * * * * * * **+ &-2222 ]+37NNNNNs$A$$A(+A(8C  CCctjdgd}|ddd}|ddd}t|||g|j\}}}t ||sJ||ksJ||usJt ||sJ||ksJdS)NMyTuple)rArJrQrrjrhr+)r namedtupler1r,r)rr:t1t2 depickled_t1depickled_MyTuple depickled_t2s r test_namedtuplezCloudPickleTest.test_namedtuples(OOODD WQ1   WQ1  8G "  9797975 ',00000r!!!! G++++,00000r!!!!!!r"z'fails sometimes for pypy on conda-forgec~d|j}ttj|dS)NaO from srsly.tests.cloudpickle.testutils import subprocess_pickle_echo CONSTANT = 42 class Foo(object): def method(self, x): return x foo = Foo() def f0(x): return x ** 2 def f1(): return Foo def f2(x): return Foo().method(x) def f3(): return Foo().method(CONSTANT) def f4(x): return foo.method(x) def f5(x): # Recursive call to a dynamically defined function. if x <= 0: return f4(x) return f5(x - 1) + 1 cloned = subprocess_pickle_echo(lambda x: x**2, protocol={protocol}) assert cloned(3) == 9 cloned = subprocess_pickle_echo(f0, protocol={protocol}) assert cloned(3) == 9 cloned = subprocess_pickle_echo(Foo, protocol={protocol}) assert cloned().method(2) == Foo().method(2) cloned = subprocess_pickle_echo(Foo(), protocol={protocol}) assert cloned.method(2) == Foo().method(2) cloned = subprocess_pickle_echo(f1, protocol={protocol}) assert cloned()().method('a') == f1()().method('a') cloned = subprocess_pickle_echo(f2, protocol={protocol}) assert cloned(2) == f2(2) cloned = subprocess_pickle_echo(f3, protocol={protocol}) assert cloned() == f3() cloned = subprocess_pickle_echo(f4, protocol={protocol}) assert cloned(2) == f4(2) cloned = subprocess_pickle_echo(f5, protocol={protocol}) assert cloned(7) == f5(7) == 7 r+rr,rrrrrs r #test_interactively_defined_functionz3CloudPickleTest.test_interactively_defined_function'sB ; v FDMF * *w x !!6!677777r"cd}dD]?}||j|}ttj|@dS)Na; from srsly.tests.cloudpickle.testutils import subprocess_pickle_echo from srsly.cloudpickle import dumps, loads def local_clone(obj, protocol=None): return loads(dumps(obj, protocol=protocol)) VARIABLE = "default_value" def f0(): global VARIABLE VARIABLE = "changed_by_f0" def f1(): return VARIABLE assert f0.__globals__ is f1.__globals__ # pickle f0 and f1 inside the same pickle_string cloned_f0, cloned_f1 = {clone_func}([f0, f1], protocol={protocol}) # cloned_f0 and cloned_f1 now share a global namespace that is isolated # from any previously existing namespace assert cloned_f0.__globals__ is cloned_f1.__globals__ assert cloned_f0.__globals__ is not f0.__globals__ # pickle f1 another time, but in a new pickle string pickled_f1 = dumps(f1, protocol={protocol}) # Change the value of the global variable in f0's new global namespace cloned_f0() # thanks to cloudpickle isolation, depickling and calling f0 and f1 # should not affect the globals of already existing modules assert VARIABLE == "default_value", VARIABLE # Ensure that cloned_f1 and cloned_f0 share the same globals, as f1 and # f0 shared the same globals at pickling time, and cloned_f1 was # depickled from the same pickle string as cloned_f0 shared_global_var = cloned_f1() assert shared_global_var == "changed_by_f0", shared_global_var # f1 is unpickled another time, but because it comes from another # pickle string than pickled_f1 and pickled_f0, it will not share the # same globals as the latter two. new_cloned_f1 = loads(pickled_f1) assert new_cloned_f1.__globals__ is not cloned_f1.__globals__ assert new_cloned_f1.__globals__ is not f1.__globals__ # get the value of new_cloned_f1's VARIABLE new_global_var = new_cloned_f1() assert new_global_var == "default_value", new_global_var ) local_cloner)r, clone_funcrC)r code_templaterHrs r *test_interactively_defined_global_variablez:CloudPickleTest.test_interactively_defined_global_variablejsb4 jD < .f0s(7%%%r"ctSrrNr)r"r rzKCloudPickleTest.test_closure_interacting_with_a_global_variable..f1s,,r"r+rM)rOr1r, __globals__r.r/rr-) r orig_valuerPr cloned_f0 cloned_f1 pickled_f1shared_global_var new_cloned_f1new_global_vars r /test_closure_interacting_with_a_global_variablez?CloudPickleTest.test_closure_interacting_with_a_global_variablesy$7777* , / 8 8 8 - - -$3B8<@M$K$K$K Iy (I,AAAAA(>>>>%*2 FFFJ IKKK)O;;;; !* $7779J777 #L44M ,I4IIIII ,BNBBBB+]__N!_444n4444$. ! ! !J ! . . . .s CC66C:cZd|j}t|dS)Naif __name__ == "__main__": from srsly.tests.cloudpickle.testutils import subprocess_worker def interactive_function(x): return x + 1 with subprocess_worker(protocol={protocol}) as w: assert w.run(interactive_function, 41) == 42 # Define a new function that will call an updated version of # the previously called function: def wrapper_func(x): return interactive_function(x) def interactive_function(x): return x - 1 # The change in the definition of interactive_function in the main # module of the main process should be reflected transparently # in the worker process: the worker process does not recall the # previous definition of `interactive_function`: assert w.run(wrapper_func, 41) == 40 r+rr,rrDs r &test_interactive_remote_function_callsz6CloudPickleTest.test_interactive_remote_function_callss5 2 FDMF * *3 4 !&&&&&r"cZd|j}t|dS)NaSif __name__ == "__main__": from srsly.tests.cloudpickle.testutils import subprocess_worker import sys with subprocess_worker(protocol={protocol}) as w: GLOBAL_VARIABLE = 0 class CustomClass(object): def mutate_globals(self): global GLOBAL_VARIABLE GLOBAL_VARIABLE += 1 return GLOBAL_VARIABLE custom_object = CustomClass() assert w.run(custom_object.mutate_globals) == 1 # The caller global variable is unchanged in the main process. assert GLOBAL_VARIABLE == 0 # Calling the same function again starts again from zero. The # worker process is stateless: it has no memory of the past call: assert w.run(custom_object.mutate_globals) == 1 # The symbols defined in the main process __main__ module are # not set in the worker process main module to leave the worker # as stateless as possible: def is_in_main(name): return hasattr(sys.modules["__main__"], name) assert is_in_main("CustomClass") assert not w.run(is_in_main, "CustomClass") assert is_in_main("GLOBAL_VARIABLE") assert not w.run(is_in_main, "GLOBAL_VARIABLE") r+r\rDs r 5test_interactive_remote_function_calls_no_side_effectzECloudPickleTest.test_interactive_remote_function_calls_no_side_effects8( P FDMF * *Q R !&&&&&r"cZd|j}t|dS)Naif __name__ == "__main__": from srsly.tests.cloudpickle.testutils import subprocess_worker with subprocess_worker(protocol={protocol}) as w: class CustomCounter: def __init__(self): self.count = 0 def increment(self): self.count += 1 return self counter = CustomCounter().increment() assert counter.count == 1 returned_counter = w.run(counter.increment) assert returned_counter.count == 2, returned_counter.count # Check that the class definition of the returned instance was # matched back to the original class definition living in __main__. assert isinstance(returned_counter, CustomCounter) # Check that memoization does not break provenance tracking: def echo(*args): return args C1, C2, c1, c2 = w.run(echo, CustomCounter, CustomCounter, CustomCounter(), returned_counter) assert C1 is CustomCounter assert C2 is CustomCounter assert isinstance(c1, CustomCounter) assert isinstance(c2, CustomCounter) r+r\rDs r 2test_interactive_dynamic_type_and_remote_instanceszBCloudPickleTest.test_interactive_dynamic_type_and_remote_instances"s8# F FDMF * *G H !&&&&&r"cZd|j}t|dS)zSimulate objects stored on workers to check isinstance semantics Such instances stored in the memory of running worker processes are similar to dask-distributed futures for instance. a if __name__ == "__main__": import srsly.cloudpickle as cloudpickle, uuid from srsly.tests.cloudpickle.testutils import subprocess_worker with subprocess_worker(protocol={protocol}) as w: class A: '''Original class definition''' pass def store(x): storage = getattr(cloudpickle, "_test_storage", None) if storage is None: storage = cloudpickle._test_storage = dict() obj_id = uuid.uuid4().hex storage[obj_id] = x return obj_id def lookup(obj_id): return cloudpickle._test_storage[obj_id] id1 = w.run(store, A()) # The stored object on the worker is matched to a singleton class # definition thanks to provenance tracking: assert w.run(lambda obj_id: isinstance(lookup(obj_id), A), id1) # Retrieving the object from the worker yields a local copy that # is matched back the local class definition this instance # originally stems from. assert isinstance(w.run(lookup, id1), A) # Changing the local class definition should be taken into account # in all subsequent calls. In particular the old instances on the # worker do not map back to the new class definition, neither on # the worker itself, nor locally on the main program when the old # instance is retrieved: class A: '''Updated class definition''' pass assert not w.run(lambda obj_id: isinstance(lookup(obj_id), A), id1) retrieved1 = w.run(lookup, id1) assert not isinstance(retrieved1, A) assert retrieved1.__class__ is not A assert retrieved1.__class__.__doc__ == "Original class definition" # New instances on the other hand are proper instances of the new # class definition everywhere: a = A() id2 = w.run(store, a) assert w.run(lambda obj_id: isinstance(lookup(obj_id), A), id2) assert isinstance(w.run(lookup, id2), A) # Monkeypatch the class defintion in the main process to a new # class method: A.echo = lambda cls, x: x # Calling this method on an instance will automatically update # the remote class definition on the worker to propagate the monkey # patch dynamically. assert w.run(a.echo, 42) == 42 # The stored instance can therefore also access the new class # method: assert w.run(lambda obj_id: lookup(obj_id).echo(43), id2) == 43 r+Nr\rDs r 9test_interactive_dynamic_type_and_stored_remote_instanceszICloudPickleTest.test_interactive_dynamic_type_and_stored_remote_instancesIs9 E J FDMF * *K L !&&&&&r"z1Seems to have issues outside of linux and CPythoncZd|j}t|dS)Naif __name__ == "__main__": from srsly.tests.cloudpickle.testutils import subprocess_worker import struct with subprocess_worker(protocol={protocol}) as w: reference_size = w.memsize() assert reference_size > 0 def make_big_closure(i): # Generate a byte string of size 1MB itemsize = len(struct.pack("l", 1)) data = struct.pack("l", i) * (int(1e6) // itemsize) def process_data(): return len(data) return process_data for i in range(100): func = make_big_closure(i) result = w.run(func) assert result == int(1e6), result import gc w.run(gc.collect) # By this time the worker process has processed 100MB worth of data # passed in the closures. The worker memory size should not have # grown by more than a few MB as closures are garbage collected at # the end of each remote function call. growth = w.memsize() - reference_size # For some reason, the memory growth after processing 100MB of # data is ~10MB on MacOS, and ~1MB on Linux, so the upper bound on # memory growth we use is only tight for MacOS. However, # - 10MB is still 10x lower than the expected memory growth in case # of a leak (which would be the total size of the processed data, # 100MB) # - the memory usage growth does not increase if using 10000 # iterations instead of 100 as used now (100x more data) assert growth < 1.5e7, growth r+r\rDs r 5test_interactive_remote_function_calls_no_memory_leakzECloudPickleTest.test_interactive_remote_function_calls_no_memory_leaks8* T FDMF * *U V !&&&&&r"cttttfD]m}t |d}t j|tjf5tj ||j dddn #1swxYwYndS)Nrr+) rr9rrdrrrr PicklingErrorr.r/r,)rexc_typer0s r test_pickle_reraisez#CloudPickleTest.test_pickle_reraises"J <H ? ?H %11C&*>?@@ ? ?!# >>>> ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?sA==B B cddi}t|j|j}||dd||dddS)NrArr+rJ)r1getr,rz)rr{depickled_methods r test_unhashable_functionz(CloudPickleTest.test_unhashable_functionsl !H*154=III ))#..222 ))#..55555r")rh zJDeprecation warning in python 3.12 about future deprecation in python 3.14cLtjdd}t|t|t||j}|||ut dD]2}|t|t|3dS)Nrrj)stepr+rx) itertoolscountnextr1r,rryrz)rcounter new_counterrzs r test_itertools_countz$CloudPickleTest.test_itertools_counts/!!,,, W  W %g FFF  {2333r ? ?A   T']]D,=,= > > > > ? ?r"cddlm}d|fd}t||j}||jjdS)NrrcdSrr)r)r"r rz=CloudPickleTest.test_wraps_preserves_function_name..f Dr"cdSrr)rsr rz=CloudPickleTest.test_wraps_preserves_function_name..g ACCCCCr"r+)r:rr1r,rzr&rrrrrs @r "test_wraps_preserves_function_namez2CloudPickleTest.test_wraps_preserves_function_names######    q      Q 7 7 7 aj11111r"cddlm}d|fd}t||j}||jjdS)NrrcdS)42Nr)r)r"r rz.fs Dr"cdSrr)r{sr rz.gr|r"r+)r:rr1r,rzrDr}s @r !test_wraps_preserves_function_docz1CloudPickleTest.test_wraps_preserves_function_docs######    q      Q 7 7 7 QY/////r"cddtd_tfd}t||j}||jjdS)NcdSrr)rs r rzDCloudPickleTest.test_wraps_preserves_function_annotations..fryr"r)rreturnc|dSrr))rrs r rzDCloudPickleTest.test_wraps_preserves_function_annotations..gs AaDDDDDr"r+)r__annotations__rr1r,rz)rrrrs @r )test_wraps_preserves_function_annotationsz9CloudPickleTest.test_wraps_preserves_function_annotationss   #$u55 q      Q 7 7 7 +Q->?????r"cltjttf}t ||ksJdSr)typingUnionrrUr1)rrs r test_type_hintzCloudPickleTest.test_type_hint s1 Ls #q!!Q&&&&&&r"cdgdfD]Gfdd}|}t||j}||fD]U}||jdt jt 5d|_dddn #1swxYwYVGdd|}|}tj ||j}~tj |}|j dksJdS) Nregistered_attributeceZdZZdZdS)@CloudPickleTest.test_instance_with_slots..ClassWithSlotscd|_dS)NrH)rr$s r r!zICloudPickleTest.test_instance_with_slots..ClassWithSlots.__init__s02D---r"N)r&r'r( __slots__r!)slotssr ClassWithSlotsrs)! 33333r"rr+rHrceZdZdZdS)CCloudPickleTest.test_instance_with_slots..SubclassWithSlotscd|_dSr)unregistered_attributer$s r r!zLCloudPickleTest.test_instance_with_slots..SubclassWithSlots.__init__"s23D///r"Nrr)r"r SubclassWithSlotsr!s#44444r"r) r1r,rzrrrrnon_registered_attributer.r/r-r)rr initial_objr7r0rrrs @r test_instance_with_slotsz(CloudPickleTest.test_instance_with_slotss-.0FG = =E 3 3 3 3 3 3 3 3 3 3 ).**K+dm555M$]3 5 5  !92>>>]>225534C0555555555555555 4 4 4 4 4N 4 4 4$#%%C!# >>>A!'-a00M 71<<<<<1 = =s2BB B MappingProxyTypez-Old versions of Python do not have this type.cjtjddi}|t||jksJdS)Nsome_keyz some valuer+)rrr1r,)rmps r test_mappingproxyz!CloudPickleTest.test_mappingproxy,s? #Z$> ? ?_R$-@@@@@@@@@r"c tjd}|ddtfg}|d}t ||j|jt ||jjcxkrdksnJdS)N dataclasses DataClassrrHrr+)rrmake_dataclassrUr1r,r)rrrr$s r test_dataclasszCloudPickleTest.test_dataclass2s)-88 ..{c3ZLII y2 DM::::v FFFHNNNNBNNNNNNNNr"cGddttj}Gfdd|tjjg|j\}}}||usJ||jusJ||jusJt|tsJ|sJ|jusJ|usJtj|j}|jusJdS)NceZdZdZdS)=CloudPickleTest.test_locally_defined_enum..StringEnumz4Enum when all members are also (and must be) stringsN)r&r'r(rDr)r"r StringEnumr<s F F F Fr"rc*eZdZdZdZdZdZfdZdS)8CloudPickleTest.test_locally_defined_enum..Colorz3-element color space123c|juSr)GREEN)rColors r is_greenzACloudPickleTest.test_locally_defined_enum..Color.is_greenEsu{**r"N)r&r'r(rDREDrBLUEr)rsr rr?sC ' 'CED + + + + + + +r"rr+) renumEnumr1rr,rrr)rrgreen1green2 ClonedColorgreen3rs @r test_locally_defined_enumz)CloudPickleTest.test_locally_defined_enum;sY G G G G Gdi G G G + + + + + + +J + + +'6 [%+u - 'G'G'G# *****[-----&#&&&&&     $$$$e#### t}EEE$$$$$$r"ctjddddd}t|j|j|g|j\}}}||usJ||jusJ||jusJ||usJdS)Nrrrjrh)rrrr+)rIntEnumr1rr,r)r DynamicColorrrClonedDynamicColors r test_locally_defined_intenumz,CloudPickleTest.test_locally_defined_intenumXs|GQA-N-NOO -<  !3\ B].$.$.$**+11111/44444!\111111r"cZd|j}t|dS)Naif __name__ == "__main__": from enum import Enum from srsly.tests.cloudpickle.testutils import subprocess_worker with subprocess_worker(protocol={protocol}) as w: class Color(Enum): RED = 1 GREEN = 2 def check_positive(x): return Color.GREEN if x >= 0 else Color.RED result = w.run(check_positive, 1) # Check that the returned enum instance is reconciled with the # locally defined Color enum type definition: assert result is Color.GREEN # Check that changing the definition of the Enum class is taken # into account on the worker for subsequent calls: class Color(Enum): RED = 1 BLUE = 2 def check_positive(x): return Color.BLUE if x >= 0 else Color.RED result = w.run(check_positive, 1) assert result is Color.BLUE r+r\rDs r test_interactively_defined_enumz/CloudPickleTest.test_interactively_defined_enumes8  @ FDMF * *A B !&&&&&r"ctjdddlm}|\}}t ||gddgD]C\}}|d|dksJt ||j}|d|dksJDdS) Nrr)relative_imports_factorymodulepackagez hello from a !r+)rrrrrr1r,)rrrrrsource cloned_funcs r $test_relative_import_inside_functionz4CloudPickleTest.test_relative_import_inside_functions2333 BAAAAA''))1A9(=>> > >LD&4666V6666666*$GGGK;==$=F$=$=$======  > >r"cddd}t||j}||fD]"}|ddksJ|jddiksJ#dS)Nr)rJc ||zSrr)rs r rzUCloudPickleTest.test_interactively_defined_func_with_keyword_only_argument..frr"r+rjrhrJ)r1r,__kwdefaults__)rrrrs r :test_interactively_defined_func_with_keyword_only_argumentzJCloudPickleTest.test_interactively_defined_func_with_keyword_only_arguments     &a$-@@@  $ 3 3D477a<<<<&3(22222 3 3r"co_posonlyargcountz(Requires positional-only argument syntaxc~d|j}ttj|dS)Na import pytest from srsly.cloudpickle import loads, dumps def f(a, /, b=1): return a + b depickled_f = loads(dumps(f, protocol={protocol})) for func in (f, depickled_f): assert func(2) == 3 assert func.__code__.co_posonlyargcount == 1 with pytest.raises(TypeError): func(a=2) r+rCrDs r =test_interactively_defined_func_with_positional_only_argumentzMCloudPickleTest.test_interactively_defined_func_with_positional_only_arguments?  FDMF * * !!6!677777r"ctjdddlm}|dksJt ||j}||usJdS)Nrr)some_singletonrr+)rrrrr%r1r,)rrdepickled_singletons r test___reduce___returns_stringz.CloudPickleTest.test___reduce___returns_stringsy 2333777777((**.>>>>>- T]444"n444444r"cd}ttj|j}|dhksJt ||j}|}|tksJdS)Nc d}|S)NctSrrNr)r"r inner_functionziCloudPickleTest.test_cloudpickle_extract_nested_globals..function_factory..inner_functions,,r"r))rs r function_factoryzQCloudPickleTest.test_cloudpickle_extract_nested_globals..function_factorys - - -" !r"rOr+)rr._extract_code_globalsrrWr1r,rO)rrglobals_depickled_factory inner_funcs r 'test_cloudpickle_extract_nested_globalsz7CloudPickleTest.test_cloudpickle_extract_nested_globalss " " " {.DD  %'''+tvv//344444+,<59]DDD&&(( z||4444444r"cGdd}|}tjtjd5t j|ddddS#1swxYwYdS)NceZdZdZdS)9CloudPickleTest.test_recursion_during_pickling..Ac"t||Sr)getattrr s r __getattribute__zJCloudPickleTest.test_recursion_during_pickling..A.__getattribute__stT***r"N)r&r'r(rr)r"r rErs# + + + + +r"rE recursionr)rrrrgr.r/)rrErAs r test_recursion_during_picklingz.CloudPickleTest.test_recursion_during_picklings + + + + + + + + ACC ]6/{ C C C ! !  a  ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !sAA!Ac|jdkrtjdtjdGfdd}|}g}t j||j|j}t|dksJtj ||}j |j |j dS) NrzNeed Pickle Protocol 5 or laternumpyc0eZdZjdZdS)ECloudPickleTest.test_out_of_band_buffers..LocallyDefinedClassrxN)r&r'r(zerosr$)rBsr LocallyDefinedClassrs28B< > >Bwwwwwwr"cptjd}t||f|j\}}||usJdSr)rrr1r,)rr depickled_T1 depickled_T2s r 'test_pickle_dynamic_typevar_memoizationz7CloudPickleTest.test_pickle_dynamic_typevar_memoizationsO N3  %4aV>Bm&M&M&M" l|++++++r"ctjdddlm}t ||j}||usJddlm}|t ||jusJdS)Nrr)rr+)AnyStr)rrrrr1r,rr)rrT1rs r test_pickle_importable_typevarz.CloudPickleTest.test_pickle_importable_typevar s2333****** Q 7 7 7Qwwww "!!!!!$-HHHHHHHHHr"cLtjd}Gddtj|}t||j|usJt|t |j|t ksJt |j5}d}t|t dddk}||t |t |dksJ|||t |t |dksJ ddddS#1swxYwYdS) NrceZdZdS),CloudPickleTest.test_generic_type..CNror)r"r rTr rqr"rTr+cz|j|usJt|jdksJ|jd}|jtjusJ|r,t|jdksJ|jd|usJn+t|jdksJ|jd|usJt|jdksJdS)Nrrok) __origin__r__orig_bases__rGeneric__args____parameters__)genericorigin type_valueuse_argsris r check_genericz8CloudPickleTest.test_generic_type..check_generic s)V33336011Q6666*1-}6666Cw/00A5555"+A.*<<<<<w566!;;;;"1!4 BBBB2,--2222tr"rr)r) rrrr1r,rUrrrun)rrrTworkerrrs r test_generic_typez!CloudPickleTest.test_generic_type s N3       q!   q4=999Q>>>>qv >>>!C&HHHH  6 6 6 O&   &qvz266".BaseNror)r"r rr%7 rqr"rceZdZdS)9CloudPickleTest.test_generic_subclass..DerivedAnyNror)r"r DerivedAnyr': rqr"r(ceZdZdS)6CloudPickleTest.test_generic_subclass..LeafAnyNror)r"r LeafAnyr*= rqr"r+ceZdZdS)9CloudPickleTest.test_generic_subclass..DerivedIntNror)r"r DerivedIntr-@ rqr"r.ceZdZdS)6CloudPickleTest.test_generic_subclass..LeafIntNror)r"r LeafIntr0C rqr"r1ceZdZdS)7CloudPickleTest.test_generic_subclass..DerivedTNror)r"r DerivedTr3F rqr"r4ceZdZdS)4CloudPickleTest.test_generic_subclass..LeafTNror)r"r LeafTr6I rqr"r7r+c:||ksJdS)Nr)mro)klass expected_mros r check_mroz8CloudPickleTest.test_generic_subclass..check_mroT s yy{{l2222tr"r) rrrrUr1r,rr9r )rrrr(r+r.r1r4r7klassesr:r!r<r9s r test_generic_subclassz%CloudPickleTest.test_generic_subclass4 s N3       6>!$                j        c        j        tAw        HQK    *gz7He  K KE"54=AAAUJJJJJ  6 6 6 A&   ! A Aiikk y,,,,,zz)UC88D@@@@@ A A A A A A A A A A A A A A A A A A AsAD11D58D5c4t|j5}tD]XGfdd}di|_d}|}||ddksJ|||ddksJY ddddS#1swxYwYdS)Nr+c eZdZddfdZdS)KCloudPickleTest.test_locally_defined_class_with_type_hints..MyClassr rc|Srr))rr s r rBzRCloudPickleTest.test_locally_defined_class_with_type_hints..MyClass.methoda s" r"Nr)type_sr rprA` s7#%#E######r"rp attributec|jd|ksJ|jjd|ksJ|jjd|ksJdS)NrDr rr)rrB)r0 expected_typeexpected_type_strs r check_annotationszUCloudPickleTest.test_locally_defined_class_with_type_hints..check_annotationse sc.{;}LLLL 259]JJJJ 28<()))) 4r"rCr)rr,_all_types_to_testrr )rr!rprHr0rCs @r *test_locally_defined_class_with_type_hintsz:CloudPickleTest.test_locally_defined_class_with_type_hints] s5  6 6 6 &+--  ##########,7*>'    gii((eW==EEEEJJ0#ugFF$NNNNN)                   sA(B  BBctjd}|j|jdfD] }t||j}||ksJ!dS)Ntyping_extensionsrAr+)rrLiteralr1r,rrLr0r7s r test_generic_extensions_literalz/CloudPickleTest.test_generic_extensions_literalv sf"/0CDD%-/@/H/MN ( (C+C$-HHHM C''''' ( (r"ctjd}|j|jtfD] }t ||j}||ksJ!dS)NrLr+)rrFinalrUr1r,rNs r test_generic_extensions_finalz-CloudPickleTest.test_generic_extensions_final| sf"/0CDD%+->-DS-IJ ( (C+C$-HHHM C''''' ( (r"cGdd}dti|_t||j}|j|jksJdS)NceZdZdS)1CloudPickleTest.test_class_annotations..CNror)r"r rTrU rqr"rTrAr+)rUrr1r,)rrTC1s r test_class_annotationsz&CloudPickleTest.test_class_annotations sg         #J Q 7 7 7!Q%6666666r"czdtdtfd}t||j}|j|jksJdS)NrArcdSrr))rAs r rz4CloudPickleTest.test_function_annotations..f ryr"r+)rUrr1r,r)rrrs r test_function_annotationsz)CloudPickleTest.test_function_annotations sW      Q 7 7 7!Q%6666666r"c"ddl} Gddfd}||j<}t||j}t |dsJ |jdS#|jwxYw)NrceZdZdS)CCloudPickleTest.test_always_use_up_to_date_copyreg..MyClassNror)r"r rpr] sr"rpcdddifS)Nr) custom_reduceTr))rrps r reduce_myclasszJCloudPickleTest.test_always_use_up_to_date_copyreg..reduce_myclass s_d$;;;r"r+r_)copyregrr1r,rr)rrar`my_objdepickled_myobjrps @r "test_always_use_up_to_date_copyregz2CloudPickleTest.test_always_use_up_to_date_copyreg s  0         < < < < </=G "7 +WYYF-ft}MMMO?O<< < << <  " & &w / / / / /G " & &w / / / /s AA22Bc`Gdd}|}t||jdS)Nc$eZdZedZdS):CloudPickleTest.test_literal_misdetection..MyClasscdS)Nr)r)r$s r __values__zECloudPickleTest.test_literal_misdetection..MyClass.__values__ srr"N)r&r'r(rrir)r"r rprg s-   X   r"rpr+r1r,rrpos r test_literal_misdetectionz)CloudPickleTest.test_literal_misdetection sR         GIIDM222222r"c`Gdd}|}t||jdS)Nc$eZdZedZdS)DCloudPickleTest.test_final_or_classvar_misdetection..MyClassctSr)rUr$s r __type__zMCloudPickleTest.test_final_or_classvar_misdetection..MyClass.__type__ s r"N)r&r'r(rrrr)r"r rprp s-   X   r"rpr+rjrks r #test_final_or_classvar_misdetectionz3CloudPickleTest.test_final_or_classvar_misdetection sR         GIIDM222222r"c tj} tjt }t tjdt tj|tjd|t|j 5}| d|ddl m ddl m m m t!jt$5| ddddn #1swxYwYt!jt$5| ddddn #1swxYwY fD]Kt!jt$5| fddddn #1swxYwYLt'| fd ksJ| fd  jksJ| fd  ksJ| fd ksJdd lm}m}m||c t!jt$5| ddddn #1swxYwYt!jt$5| fddddn #1swxYwYt5t!jt$5| fddddn #1swxYwYt!jt$5| ddddn #1swxYwYddl}t'|| fdksJ| fd ksJt5|t!jt$5| fddddn #1swxYwYt!jt$5| fddddn #1swxYwYddl}t'|jj| fd ksJ| fd jksJ| fd ksJt!jt$5| fddddn #1swxYwYt!jt$5| ddddn #1swxYwYt5|jjt!jt$5| fddddn #1swxYwYddl}t'|j| fd ksJ| fd jksJ| fd ksJt5|jdddn #1swxYwYd}|t_||d |d!|d"fD]H}tj |dr$jtCvrt5IdS#d}|t_||d |d!|d"fD]H}tj |dr$jtCvrt5IwxYw)#Nrr+c@tj|Sr)rrr8)ps r rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value.. schooa00r")local_functionLocalT LocalClassc tdSNzmock_local_folder.mod __import__r)r"r rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value.. *-D"E"Er"c tdSNzmock_local_folder.subfolder.modr}r)r"r rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value..  +L M Mr"cSrr))rlsr rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value.. sar"cSrr)rxsr rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value.. ^^%5%5r"cjSrr&)rysr rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value.. s V_r"c<SrrB)rzsr rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value.. s**,,"5"5"7"7r"c,Srr)rsr rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value.. s#"4"4"6"6r")LocalSubmodClass LocalSubmodTlocal_submod_functionc tdSrr}r)r"r rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value.. rr"cSrr)rsr rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value.. "7r"cSrr)rsr rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value.. r'r"c tdSr|r}r)r"r rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value.. rr"cSrr)rsr rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value.. rr"cSrr)_funcsr rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value.. UUWWr"cSrr)rsr rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value.. r'r"cSrr)rsr rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value..! rr"cSrr)rsr rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value..' rr"cjSrr_tsr rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value..( R[r"c<Srr_classsr rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value..) VVXX__%6%6r"cSrr)rsr rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value../ r'r"c tdSr|r}r)r"r rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value..1 rr"cSrr)rsr rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value..7 rr"cSrr)rsr rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value..< rr"cjSrrrsr rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value..= rr"c<Srrrsr rzeCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value..> rr"mock_local_folderz.modz .subfolderz.subfolder.submod)"rrcopyrdirname__file__r<insertrr,r mock_local_folder.modrrxryrzrrrr r&rB"mock_local_folder.subfolder.submodrrrr r subfoldersubmodmock_local_folder.subfolderrrr )r_prev_sys_path_mock_interactive_session_cwdwrrr_fnamerrzryrrrrxrrrls @@@@@@@@@r Ctest_pickle_constructs_from_module_registered_for_pickling_by_valuezSCloudPickleTest.test_pickle_constructs_from_module_registered_for_pickling_by_value s N 4-/GOOH,E,E ) #(B ' ' ' #($A B B B HOOA< = = ="DM:::s Ha002O433333 ];//GGEEEEFFFGGGGGGGGGGGGGGG];//EEMM ~vzB))A{33))iiii((())))))))))))))))---uu555566..:J:JJJJJuu444455HHHHEE777788!z||**,,---- EE666677>>;K;KKKKK !"79I"E6];//EEMM];//99EE7777888999999999999999+3///];//22EE0000111222222222222222];//GGEEEEFFFGGGGGGGGGGGGGGG)(((():;;;uu555566..:J:JJJJJuu____--8888*+<===];//22EE0000111222222222222222];//99EE7777888999999999999999 :999():)D)KLLLuu____--8888uu000011R[@@@@uu6666776688??;L;LLLLL ];//22EE0000111222222222222222];//GGEEEEFFFGGGGGGGGGGGGGGG+%/6];//99EE77778889999999999999993222():)DEEEuu____--8888uu000011R[@@@@uu6666776688??;L;LLLLL*+<+FGGGgs Hs Hs Hs Hs Hs Hs Hs Hs Hs Hs Hs Hs Hs Hs Hj)F%CHooo&/D/D/D2224 4 4kooa..43<+H+J+JJJ.s333  4 4)F%CHooo&/D/D/D2224 4 4kooa..43<+H+J+JJJ.s333  4sB ]06A[=7D [=D [=!D "[=>E! [=!E% %[=(E% )$[= F2& [=2F6 6[=9F6 :C$[=K5 [=K [=K [=%L > [= L [=L +[==M" [="M& &[=)M& *[=N) [=)N- -[=0N- 1B [=;Q  [= Q$ $[='Q$ ([=R) [=)R- -[=0R- 1B:[=+V [=V [=V [=4W [=W [=W 5[=X9- [=9X= =[=X= B0[=1 ]0=\]0\]00A(_ctjddD]L}|dkrddl}|j|j}n|dkrddlm}|j|j} t|j 5}| fd|ksJt|| fd|ksJd |_|j|ksJ| fd d ksJt|dddn #1swxYwY||_|j tvrt|#||_|j tvrt|wwxYwdS) Nr)rrrrrr+cSrr)r{sr rzqCloudPickleTest.test_pickle_constructs_from_installed_packages_registered_for_pickling_by_value..Y r"cSrr)r{sr rzqCloudPickleTest.test_pickle_constructs_from_installed_packages_registered_for_pickling_by_value.._ rr"modified globalcSrr)r{sr rzqCloudPickleTest.test_pickle_constructs_from_installed_packages_registered_for_pickling_by_value..b rr")rrrpackage_function_with_globalglobal_variablerrmodule_function_with_globalrr,r r r r&r )rpackage_or_moduler_original_globalrrs @r Otest_pickle_constructs_from_installed_packages_registered_for_pickling_by_valuez_CloudPickleTest.test_pickle_constructs_from_installed_packages_registered_for_pickling_by_valueJ s 2333!6 2 2  I--00002#$#4  "h..4444441#$#4  2& >>> 2!55--1AAAAA -Q///55--1AAAAA(9A%,0@@@@@55--1BBBBB.q111 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2%5!:!>!@!@@@.q111%5!:!>!@!@@@.q1111A1 2 2s1D5$B C=1 D5=D D5D D55.E#ctjdddl}ddlm}|j}d}d}d} t |j5}|||||dt|d |_|||d ||d|ksJ||d d ksJ dddn #1swxYwY||_dtvrt|dSdS#||_dtvrt|wwxYw) Nrr)rcHtdjd}i|_dSNr__main__r~r_cloudpickle_registry)_mains r _create_registryzzCloudPickleTest.test_pickle_various_versions_of_the_same_function_with_different_pickling_method.._create_registryu s%u%%-j9E*,E ' ' 'r"cNtdjd}||j|<dSrr)vkrs r _add_to_registryzzCloudPickleTest.test_pickle_various_versions_of_the_same_function_with_different_pickling_method.._add_to_registryz s*u%%-j9E-.E ' * * *r"c`tdjd}|j|Srr)rrs r _call_from_registryz}CloudPickleTest.test_pickle_various_versions_of_the_same_function_with_different_pickling_method.._call_from_registry~ s-u%%-j9E15.q133 3r"r+f_by_refrf_by_val) rrrrrrr,r r r r )rrrrrrrrs r Ptest_pickle_various_versions_of_the_same_function_with_different_pickling_methodz`CloudPickleTest.test_pickle_various_versions_of_the_same_function_with_different_pickling_methodi s 2333 $###JJJJJJ/? - - -  / / / 4 4 4 A"DM::: a&'''&:666()=>>>7H$4&:666EE-z::>NNNNNEE-z::>OOOOOO               4D 0%)F)H)HHH*+?@@@@@IH4D 0%)F)H)HHH*+?@@@@Is0DBC$ D$C((D+C(,D)E)rhr3z2Determinism can only be guaranteed for Python 3.7+c bt}d}tdD];}|t||jdt |i.func_with_globals s(+AA Ar"rPYTHONHASHSEED)r,add_envrz1Expected a single deterministic payload, got %d/5) rryrCrr,rr pickletoolsdisrfail)rvalsrrVvals r ,test_deterministic_pickle_bytes_for_functionzrJrOrRrWrZrdrmrsrrrrr)r"r r[r[ysT +HGGG### [ *H * , , 9 M   * Ks/?)/KHJJII JJI$ 0 0 0000* [9::%%;:% "   /// ??? AAA @@@ +++ --- ,,, /// /// IIILLL %%%"   "$ ! ! !(((4CCC&,,,,,,\...  [6688FBCEEEE$,,, ,,, / / /,,,$,,,***333///#)#)#)J)))@,,,\E5E5E5N   0828282tLLL333***...444 G G G000*???. [    y ( LS-=bqb-AW-L /-x-//69 / !$/RC4D 4R /#w. @@$F$F@@$FL<<<( [&&((F2ACCCCCCC0 [4=??!!??!, [c.FFFF1FFFFF?$$ $4   0   &DDD*DDD,$$$    666,;;; 888 ;;;4Y=Y=Y=vX=X=X=tAAA.2/2/2/h&<&<& > > 3 3 3 [GGEN4HIIIIKK88KK80555555 !!!KKK BBB ,,, III#O#O#OJ'A'A'AR2((( ((( 777777000$333333 [9::P4P4;:P4d222>,A,A,A\ [ 6!C r"r[ceZdZdZdS)Protocol2CloudPickleTestrjN)r&r'r(r,r)r"r rr sHHHr"rcbtjd}t||j}|JdS)Nrr0)rrrr&)rmodule_and_names r /test_lookup_module_and_qualname_dynamic_typevarr s7sA1!!*EEEO  " " " " "r"ctjdddl}|j}t ||j}|J|\}}||usJ|dksJdS)Nrrrr)rrrrrr&)rrrrr0s r 2test_lookup_module_and_qualname_importable_typevarr st .///A1!!*EEEO  & & &"LFD ) ) ) ) ) 3;;;;;;r"cttjtjj}|J|\}}|tusJ|dksJdS)Nrr)rrrr&)rrr0s r .test_lookup_module_and_qualname_stdlib_typevarr sb1&-7=}7MOOOO  & & &"LFD V     8      r"ctjdddl}ddlm}t t ksJt|t |jhksJt|t |j|jhksJt|t |jhksJd|jd}tj t|5t|jdddn #1swxYwYt|t t ksJ|d}tj ttj |5t|dddn #1swxYwYd|jd}tj t|5t|jdddn #1swxYwYtjd}|d}tj ttj |5t|ddddS#1swxYwYdS) Nrrz%Input should be a module object, got z insteadrz& is not registered for pickle by value dynamic_modzN was not imported correctly, have you used an `import` statement to access it?)rrrrrr rr r&r rr9reescaperr)pkgrrrs r test_register_pickle_by_valuer s[ .///&&&&****** ( * *cee 3 3 3 3S!!! ( * *s|n < < < <S!!! ( * *s|S\.J J J J Js### ( * *s|n < < < < H#, H H HC z - - -11"3<000111111111111111s### ( * *cee 3 3 3 3  8 8 8C z3 8 8 8(("3'''((((((((((((((( I#, H H HC z - - -// ...///////////////"=11K  , , , z3 8 8 8.. ---..................sH&DD D 3FFFG!!G%(G%2IIIc*tjd}Gddtj|}||t|tjtjtjtjtjttj|tjtgtjftjdtjftjgtjftjtdftjt|tftj ttj ttftj tj |ttj g}|S)NrceZdZdS)_all_types_to_test..CNror)r"r rTr s r"rT.)rrrrUAnyOptionalrCallableTupleListDictrClassVarNoReturn)rrT types_to_tests r rIrI ssA     FN1     1S6 6:v qvz)*VZ(FJ' S#X S!C&[! C CH#M" r"chttdsJtjtjusJdS)NPickler)rr.rrr)r"r test_module_level_picklerr  s7 ; * *** *  +": : : : : : :r"r)LrrqrrRr:rLrqrmrrboperatorrrrrr4rrdrUrr_rrrrrrrrrrrrB scipy.specialspecialrDrrdrrr.rrr r r srsly.cloudpickle.cloudpickler r rrrr testutilsrrrrrOrrrr1r6r<rYTestCaser[rrrrrrIr r&mainr)r"r rs    ++++++++     \" B CCCNNNNGGG('''''++++++666666888888;;;;;;EEEEEEDDDDDDDDKKKKKKKKEEEEEE------////////////((((((((#.">CCCC/// 6664v(v(v(v(v(h'v(v(v(rQ ### %.%.%.P4;;; zHMOOOOOs$8 B BBBB$#B$