d-2dZddlZddlZddlZddlZddlZddlmcm Z ddl m Z m Z ej ej ejej eZdZGddeZGddeZGd d eZejd Zejd Zejd ZejdZejdZejdZejdZ ejdZ!dZ"ej#$e j%duddZ&ej#$e dej#'dddgdZ(dZ)dZ*ej#$e j%duddZ+ej#'dgd d!d"g#d$Z,e-d%krej.dSdS)&z Tests for iofuncs.py. N)is_text_stringPY2c ||dSd}t|D]} |o0ttj||||k}nE#t $r8|o2t dt||||D}YnwxYwtt|dzt|z|S)NTcDg|]\}}tj||kS)npall).0obj1obj2s Glib/python3.11/site-packages/spyder_kernels/utils/tests/test_iofuncs.py z(are_namespaces_equal..-s<JJJ*$ %%JJJz: ) sortedkeysboolrmean ValueErrorr zipprintstr)actualexpected are_equalvars r are_namespaces_equalr#s ~(*tIhmmoo&&00 K!/d27 ,,.,.'/'/II K K K!KcJJHSMs)77rr-ceZdZdZdZdS)UnPickleableObjz8A class of objects that can deepcopied, but not pickled.c|j|j}|j|j|Sr ) __class____new__r%update)r"memonew_ones r __deepcopy__zUnPickleableObj.__deepcopy__Fs7.((88 ...rN)r(r)r*r+r9rrr r2r2Ds)BBrr2c>d}d}tjd}dtjdtjd}tjdtjd tjdtjtjdt g}|||||d S) z Define spydata file ground truth values. The file export_data.spydata contains five variables to be loaded. This fixture declares those variables in a static way. hamT)dtype)abgE@ABCDE)reye complex128int64bool_objectrCs r spydata_valuesrNOs A A q A77788A  " " "D"&"(*C*C*CRVAU[E\E\E\]A 3 33rc tjtd}t j|d}|d}|d}|d}|d}|d}|||||d S) aZ Load a Numpy pickled file. The file numpy_data.npz contains six variables, each one represents the expected test values after a manual conversion of the same variables defined and evaluated in MATLAB. The manual type conversion was done over several variable types, such as: Matrices/Vectors, Scalar and Complex numbers, Structs, Strings and Cell Arrays. The set of variables was defined to allow and test the deep conversion of a compound type, i.e., a struct that contains other types that need to be converted, like other structs, matrices and Cell Arrays. znumpy_data.npzT) allow_picklerDrErFrGrHrC)ospathjoinLOCATIONrloaditem)rRfile_srDrErFrGrHs r real_valuesrX_s 7<<"2 3 3D WT - - -Fs As As As As A 3 33rctj|}d|d<t|d<t|d<tjj|d<td|d<td|d <td |d <|S) z Define a dictionary of objects of a variety of different types to be saved. This fixture reprisents the state of the namespace before saving and filtering out un-deep-copyable, un-pickleable, and uninteresting objects. zOSome objects could not be saved: undeepcopyable_instance, unpickleable_instanceexpected_error_string module_obj class_obj function_objspamunpickleable_instancer<undeepcopyable_instanceeggscustom_instance) copydeepcopyio ExceptionrQrRrSr2r-rrNnamespace_dicts r namespace_objects_fullriws]>22N 9*+$&N< "+N;%'W\N>".=f.E.EN*+0A%0H0HN,-(1&(9(9N$% rcRtj|}td|d<|S)z Define a dictionary of the objects from the namespace that can be saved. This fixture reprisents the state of the namespace after saving and filtering out un-deep-copyable, un-pickleable, and uninteresting objects. rarb)rcrdrrgs r namespace_objects_filteredrks,]>22N(1&(9(9N$% rcLi}d|d<t|d<td|d<|S)z; Define a dictionary of that cannot be deepcopied. No supported objects to saverZr\r<r`)rfr-rhs r namespace_objects_nocopyableros7 N.LN*+"+N;0A%0H0HN,- rc`i}d|d<tjj|d<td|d<|S)z@ Define a dictionary of objects that cannot be pickled. rmrZr]r^r_)rQrRrSr2rns r namespace_objects_nopickleablerqs; N.LN*+%'W\N>".=f.E.EN*+ rcH|jdS||jSr paramgetfixturevaluerequests r input_namespacerx%}t&&w}555rcH|jdS||jSr rsrvs r expected_namespacer{ryrctjtd}t j|}t |tsJ|\}}|dtj dkr|rJdS)z6 Test the load of .npz files as dictionaries. zimport_data.npzval1r;N) rQrRrSrTiofuncs load_array isinstancetuplerarray)filenamer! variableserrors r test_npz_importrssw||H&788H  h ' 'D dE " """"Iu V  + +E + + + + +rzSciPy required)reasonc8tjtd}t j|\}}d}t |D]5}|o0ttj ||||k}6|sJdS)a Test the automatic conversion and import of variables from MATLAB. This test loads a file stored in MATLAB, the variables defined are equivalent to the manually converted values done over Numpy. This test allows to evaluate the function which processes the conversion automa- tically. i.e., The automatic conversion results should be equal to the manual conversion of the variables. zdata.matTN) rQrRrSrTr~ load_matlabrrrrr)rXrRinf_validrs r test_matlab_importrs 7<<* - -D   & &FC Ek&&(())FFE$rw{3'73s8'CDDEE LLLLLrzFails on Python 2spydata_file_namezexport_data.spydatazexport_data_renamed.spydatactjt|}t j|\}}|Jt ||sJdS)a* Test spydata handling and variable importing. This test loads all the variables contained inside a spydata tar container and compares them against their static values. It tests both a file with the original name, and one that has been renamed in order to catch Issue #9 . N)rQrRrSrTr~load_dictionaryr)rrNrRr!rs r test_spydata_importrsS 7<<"3 4 4D)$//KD% === n 5 555555rctj}tjtd}t j|\}}|rt|sJ|Jtj|ksJdS)z Test that import fails gracefully with a fn not present in the namespace. Checks that the error is caught, the message is passed back, and the current working directory is restored afterwards. z export_data_withfunction.spydataN)rQgetcwdrRrSrTr~rr) original_cwdrRr!rs r test_spydata_import_witherrorrsx9;;L 7<<"D E ED)$//KD% *^E***** <<< 9;;, & & & & & &rctj}tjtd} t j|J#t$rYnwxYwtj|ksJdS)zS Test that import fails properly when file is missing, and resets the cwd. z$non_existant_path_2019-01-23.spydataFN)rQrrRrSrTr~rIOError)rrRs r test_spydata_import_missing_filers9;;L 7<<"H I ID%%%       9;;, & & & & & &sA AActj}d|_|ddksJd|jd<|jjdksJ|ddiddksJgd|d<t j}tj|||dtj |\}}|J|ddksJ|d jdksJ|d gdgksJdS) z%Test support for matlab stlye struct.r^rArad)crA)r;rBr=rNr) r~ MatlabStructrArrreBytesIO save_matlabseekrtolist)r@bufr!rs r test_matlabstructrs A AC S6V    ACH 35F???? sFm&11 1 1 1 1 YYAcF *,,C 3HHQKKK%c**KD% === 9     9;& 9    ))) , , , , , ,rz+input_namespace,expected_namespace,filename))rNrNexport_data_copy)rirk export_data_2)roNexport_data_none_1)rqNexport_data_none_2rxr{)indirectctjt|dz}d}d|vr |d}|d=tj} t j||}||ksJ|"tj|rJnot j|\}}|Jt| t| t||sJ|tjksJ tj|r5 tj |dS#tttf$rYdSwxYwdS#tj|r3 tj |w#tttf$rYwwxYwwxYw)z Test spydata export and re-import. This test saves the variables in ``spydata`` format and then reloads and checks them to make sure they save/restore properly and no errors occur during the process. z.spydataNrZ)rQrRrSrTrr~save_dictionaryisfilerrrrremoverOSErrorPermissionError) rxr{rrRexpected_error cwd_original export_error data_actual import_errors r test_spydata_exportr-s 7<<(Z"7 8 8DN/11()@A 3 49;;L.EE ~----  %w~~d++ + + + +(/(?(E(E %K''' +""$$ % % % $))++ , , ,' 5GHH H H Hry{{***** 7>>$     $Wo6      27>>$     $Wo6     sCC E.8EE)(E). GF$#G$F>;G=F>>G__main__)/r+rerQrcpytestnumpyrspyder_kernels.utils.iofuncsutilsr~spyder_kernels.py3compatrrrRrealpathrSrdirname__file__rTrrMrr-r2fixturerNrXrirkrorqrxr{rmarkskipifrr parametrizerrrrrr(mainrrr rs   /........88888888 7  BGLL)+)B)BDD E E    ////////  ' 4 4 4444.*         666666:::G'4/8HIIJI$C 344,/D/L/NOO 6 6OO54 6 ' ' ' ' ' ' G'4/8HII--JI-*FIII #$89 ;; !! ;; !H zFKMMMMMr