o Zh-@sdZddlZddlZddlZddlZddlZddlmm Z ddl m Z m Z ej ej eej eZddZGdddeZGdd d eZGd d d eZejd d ZejddZejddZejddZejddZejddZejddZ ejddZ!ddZ"ej#j$e j%duddd d!Z&ej#j$e d"dej#'d#d$d%gd&d'Z(d(d)Z)d*d+Z*ej#j$e j%duddd,d-Z+ej#j'd.gd/ddgd0d1d2Z,e-d3kre.dSdS)4z Tests for iofuncs.py. N)is_text_stringPY2c Cs|dur |dur dSd}t|D];}z|o#tt||||k}Wnty@|o=tddt||||D}Ynwtt |dt |q|S)NTcSsg|] \}}t||kqS)npall).0Zobj1obj2rr^/var/www/html/lang_env/lib/python3.10/site-packages/spyder_kernels/utils/tests/test_iofuncs.py -sz(are_namespaces_equal..z: ) sortedkeysboolrmean ValueErrorrzipprintstr)actualexpectedZ are_equalvarrrr are_namespaces_equal#s"  rc@s eZdZdZddZddZdS) CustomObjz&A custom class of objects for testing.cCsd|_|r ||_dSdSN)data)selfrrrr __init__5s zCustomObj.__init__cCs |j|jkSr)__dict__)rotherrrr __eq__:s zCustomObj.__eq__N)__name__ __module__ __qualname____doc__rrrrrr r3s rc@eZdZdZddZdS)UnDeepCopyableObjz-A class of objects that cannot be deepcopied.cCstr) RuntimeError)rrrr __getstate__@szUnDeepCopyableObj.__getstate__N)rr r!r"r&rrrr r$> r$c@r#)UnPickleableObjz8A class of objects that can deepcopied, but not pickled.cCs |j|j}|j|j|Sr) __class____new__rupdate)rmemoZnew_onerrr __deepcopy__FszUnPickleableObj.__deepcopy__N)rr r!r"r-rrrr r(Dr'r(cCsfd}d}td}dtjdtjdd}tjdtjdd tjdtjdtjdtdg}|||||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)Zdtype)abgE@ABCDE)reyeZ complex128Zint64Zbool_objectr5rrr spydata_valuesOs  .r=cCs\tjtd}tj|dd}|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)Z allow_pickler6r7r8r9r:r5)ospathjoinLOCATIONrloaditem)r?Zfile_sr6r7r8r9r:rrr real_values_s  rDcCsVt|}d|d<t|d<t|d<tjj|d<td|d<td|d <t d |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_stringZ module_obj class_obj function_objspamunpickleable_instancer/undeepcopyable_instanceeggscustom_instance) copydeepcopyio Exceptionr>r?r@r(r$rr=namespace_dictrrr namespace_objects_fullws     rScCst|}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. rKrL)rMrNrrQrrr namespace_objects_filtereds  rTcCs$i}d|d<t|d<td|d<|S)z; Define a dictionary of that cannot be deepcopied. No supported objects to saverErFr/rJ)rPr$rRrrr namespace_objects_nocopyables  rWcCs(i}d|d<tjj|d<td|d<|S)z@ Define a dictionary of objects that cannot be pickled. rUrErGrHrI)r>r?r@r(rVrrr namespace_objects_nopickleables   rXcC|jdurdS||jSrparamZgetfixturevaluerequestrrr input_namespace  r^cCrYrrZr\rrr expected_namespacer_r`cCsLtjtd}t|}t|tsJ|\}}|dt dkr"|r$JdS)z6 Test the load of .npz files as dictionaries. zimport_data.npzZval1r.N) r>r?r@rAiofuncsZ load_array isinstancetuplerarray)filenamer variableserrorrrr test_npz_imports  rhzSciPy required)reasoncCs\tjtd}t|\}}d}t|D]}|o&tt ||||k}q|s,JdS)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) r>r?r@rAra load_matlabr r r rr)rDr?inf_validrrrr test_matlab_imports    rnzFails on Python 2spydata_file_namezexport_data.spydatazexport_data_renamed.spydatacCs:tjt|}t|\}}|dusJt||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)r>r?r@rAraload_dictionaryr)ror=r?rrgrrr test_spydata_imports  rqcCsTt}tjtd}t|\}}|rt|sJ|dus Jt|ks(JdS)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)r>getcwdr?r@rArarpr) original_cwdr?rrgrrr test_spydata_import_witherrors  rtcCsPt}tjtd}zt|Wn tyYnwJt|ks&JdS)zS Test that import fails properly when file is missing, and resets the cwd. z$non_existant_path_2019-01-23.spydataFN)r>rrr?r@rArarpIOError)rsr?rrr test_spydata_import_missing_files rvcCst}d|_|ddksJd|jd<|jjdksJ|ddiddks'Jgd|d<t}t|||dt |\}}|dusIJ|ddksQJ|d jdksZJ|d gdgksgJdS) z%Test support for matlab stlye struct.rHr3rKd)cr3)r.r4r0rNrx) raZ MatlabStructr3rxrwrOBytesIOZ save_matlabseekrjtolist)r2bufrrgrrr test_matlabstructs     r}z+input_namespace,expected_namespace,filename))r=r=Zexport_data_copy)rSrTZ export_data_2)rWNZexport_data_none_1)rXNZexport_data_none_2)Zindirectc Cs,tjt|d}d}d|vr|d}|d=t}z`t||}||ks'J|dur4tj|r3Jn t|\}}|dusAJt | t | t ||sTJ|tks\JWtj|ryzt |WdSt ttfyxYdSwdStj|rzt |Wwt ttfyYwww)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.spydataNrE)r>r?r@rArrraZsave_dictionaryisfilerprr rremoveruOSErrorPermissionError) r^r`rer?Zexpected_errorZ cwd_originalZ export_errorZ data_actual import_errorrrr test_spydata_export-s<      r__main__)/r"rOr>rMZpytestnumpyrZspyder_kernels.utils.iofuncsutilsraZspyder_kernels.py3compatrrr?realpathr@rrdirname__file__rArr<rr$r(Zfixturer=rDrSrTrWrXr^r`rhmarkZskipifrjrnZ parametrizerqrtrvr}rrmainrrrr sd                $