
    Џkh@                        d dl Z d dlmZ d dlmZmZ d dlmZ d dlm	Z	 d dl
Z
d dlmZ d dlZd dlZd dlZd dlmZmZmZmZmZmZ d dlZd dlmZ d dlZd d	lmZ d d
lmZmZm Z m!Z! d dl"Z#d dl$m%Z%m&Z&m'Z' d dl(mc m)c m*Z+ d dl,m-Z-m.Z.m/Z/m0Z0m1Z1 d dl2m3Z3m4Z4m5Z5m6Z6 d dl7m8Z8m9Z9m:Z:m;Z;m<Z< d dl=mc m)c m>Z?  e ee@      d      ZAej                  j                  ZDd ZEej                  dz   ej                  deH      j                  dd      z  ZJdddideJidgZKeKj                  dddid edg      id       eKj                  dddid ej                  eJ      d ej                  eJ      z  z   id        ej                  d      ZP eQ eRdd             ePd <    eQ eRdd            ePddd f<   eKj                  d!d"did"ePid       eKj                  d#d$d#id$ e eP      id       ePj                  eT      ZUeUd%xx   dz  cc<   eKj                  d&d'd#id' e eU      id       eKj                  d(ddd)eJePd)d       eKj                  d*d+did+ eEd,      id       eKj                  d-d.did. ed/g      id        eEg  eEdg       eEdd0gg       eEg d1g      feV      j                  dd,      ZW ed2g      eWd%<   d3d4d3id4eWidgZX eE eEd       eEd0       eEg        eEg        eEd5      feV      j                  dd,      ZY ej                  d6eV      Z[ eEd      e[d%<   eXj                  d7d8d3id8e[id       eXj                  d9d:d3id:eYid       eXj                  d;d<did< eg d=      id       eXj                  d>d?did? ej                   ej                   eQ eRdd@            dA            id        e ej                  d0       ej                  d      ej                  g      j                  dd5      Z_dBD  cg c]  } | eVf c} Z` ej                  d6e`      Za edCg      eadD   d%<   e_eadE   d%<   e_dFz  eadG   d%<   eXj                  dHdIdHidIeaid        ej                  dJeV      Zb eEd      ebd%<    ej                  dKeV      ebdL<    eEd0ej                        ebdL   d%<    eEd5ggej                        ebdL   dL<    ej                  dJeV      ebdL   dM<    eEdej                        ebdL   dM   d%<    eEdNej                        ebdL   dM   dL<   eXj                  dOdPd3idPebid        ej                  d6dQD  cg c]  } | eVf c}       Zd eEd      edd%   dR<    ej                  d6dSeVfg      edd%   dT<    edUg      edd%   dT   d%   dS<   eXj                  dVdWdHidWedid        ej                  dJdQD  cg c]  } | eVf c}       Ze eEd      eed%   dR<    eEd0      eed%   dT<    edXg      eedL   dR<    edYg      eedL   dT<   eXj                  dZd[dHid[eeid        ej                  d\D  cg c]  } | eVf c}       Zf e' ej                  d6ef      d]      Zgegd%   Zh ed^g      ehd_<    ed`g      ehda<    ed^g      ehdb<    eEd       ehdc<    eEd      ehdd<    eEd      ehde<   eXj                  dfdgdfidgegid        ei eeAdh      di      Zjejj                         j                  dj      Zmejj                          eXj                  dkdldidl eemg      id       eXj                  d#d$d#id$ e eP      id       eXj                  d&d'd#id' e eU      id       eXj                  dmdndoidn edpgdqgg      id       eXdd Zoeoj                  drdsdfids ej                  egd0      j                  dd0      id       dt Zqdu Zrdv Zsdw Ztdx ZuddyZvej                  j                  dedz      d{        Zxej                  j                  dedz      d|        Zyej                  j                  d}d~d~dgfddgfg      d        Zzd Z{d Z|d Z}d Z~d Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zej                  j                  dg d      d        Zd Zd Zd Zd Zd Zd ZdÄ ZdĄ Zdń ZdƄ Zyc c} w c c} w c c} w c c} w )    N)OrderedDict)joindirname)glob)BytesIO)mkdtemp)assert_array_equalassert_array_almost_equalassert_equalassert_assert_warnsassert_allclose)raises)array)issparse	eye_array	coo_array	csc_array)MatlabOpaqueMatlabFunctionMatlabObject)matdimsMatWriteErrorMatReadErrormatfile_versionMatWriteWarning)mat_reader_factoryloadmatsavematwhosmat)MatFile5WriterMatFile5Readervarmats_from_matto_writeableEmptyStructMarkerdatac                  P    t        j                  | i |}t        |      |_        |S )z;Convenience function to return matlab-compatible 2-D array.)npr   r   shape)argskwargsarrs      Z/var/www/teggl/fontify/venv/lib/python3.12/site-packages/scipy/io/matlab/tests/test_mio.pymlarrr.   &   s%    
((D
#F
#CCIJ       	   dtype   double
testdouble)nameclassesexpectedstring
teststringcharz+"Do nine men interpret?" "Nine men," I nod.complextestcomplexy              ?)         matrix
testmatrixsparse
testsparser   r   sparsecomplextestsparsecomplexmulti)thetaaminus	testminusonechartestonecharr   )r4   rR   r?   z@This cell contains this string and 3 arrays of increasing lengthcelltestcellr?   r4   r4   
scalarcelltestscalarcell	emptycelltestemptycellstringarrayteststringarray)zone  ztwo  three3dmatrixtest3dmatrix   )r0   r?   rR   )stringfielddoublefieldcomplexfieldzRats live on no evil star.r`   ra   y      ?      ?rb   struct
teststructr4   rR   )r4   r?   r   r4   )r   rR   r@   cellnesttestcellnestonetworj   r\   rk   znumber 3
structnestteststructnestznumber 1znumber 2	structarrteststructarr)expr	inputExprr*   isEmptynumArgsversioninlinexrp   z x = INLINE_INPUTS_{1};rq   r*   rr   rs   rt   object
testobjectzjapanese_utf8.txtrbzutf-8unicodetestunicodebool	testboolslogicalTFobjectarraytestobjectarrayc                 *   t        |       }t        |      }||u ry|t        j                  u r*| j                  dk(  rt        | j	                               |u S |t        j                  u r*|j                  dk(  rt        |j	                               |u S y)zkCheck if types are same or compatible.

    0-D numpy scalars are compatible with bare python scalars.
    T F)typer(   ndarrayr)   item)var1var2type1type2s       r-   types_compatibler      s}    
 JEJE~

tzzR/DIIK E))

tzzR/DIIK E))r/   c           	         t        |      r@t        t        |             t        |j                         |j                         | d       yt        t	        ||      dt        |       dt        |       d|         t        |t        j                  t        j                  z  t        z        st        ||       yt        |j                  |j                  k(  d|j                   d|j                   d|         |j                  }|j                  r\t        |t              r t        |j                  |j                         t!        |      D ]  \  }}|  d	| d
}t#        ||||           y|j$                  r-|j$                  D ]  }|  d| d}t#        |||   ||           y|j
                  t&        t        j(                  t        j*                  fv rt        |||        yt        ||| d       y)z/ Check one level of a potentially nested array r@   )err_msgdecimalNzExpected type z, got z at zExpected shape )msgz, [z], z, field z, r   )r   r   r
   toarrayr   r   
isinstancer(   voidr   r   r   r)   r3   	hasobject	classname	enumerate_check_levelfieldsstrstr_bool_)labelr9   actualex_dtypeievlevel_labelfns           r-   r   r      s    !!&.."2"*"2"2"4*/*+	- 	Xv.T(^,F4<.UGLN h"** 4| CDXv&HNNfll*!(..!1~T%QS~~Hh-++V-=-=>x( 	5EAr"G3qc-Kb&)4	5 	// 	3B"G8B4r2K!"vbz3	3 	}}# # 	VXu5fhqIr/   c                     |D ]X  }t        |dd      }d|  d| }|j                         D ]-  \  }}| d| }t        ||v d|        t        ||||          / Z y )NTFstruct_as_recordspmatrixtest ; file z, variable zMissing key at )r   itemsr   r   )	r7   filescase	file_namematdictr   kr9   k_labels	            r-   _load_check_caser   "  s     8	)dUKvWYK0::< 	8KAx{1#.GALOG9"=>(GAJ7	88r/   c           
         |D ]|  }d|  d| }t        |      }|j                         D cg c]  \  }}||j                  ||   f }	}}|j                          |	j                          t	        ||	| d|d|	       ~ y c c}}w )Nr   r   z: z != )r    r   r)   sortr   )
r7   r   r   r8   r   r   whosr   r9   expected_whoss
             r-   _whos_check_caser   ,  s     	vWYK0y! @Dzz|M0;8Q
+M M 			T=wb],=>	
Ms   Bc                 r    t               }t        |||       |j                  d       t        | |g|       y )Nformatr   )r   r   seekr   )r7   r9   r   
mat_streams       r-   _rt_check_caser   =  s/    JJ0OOATJ<2r/   c              #   "  K   | dk(  rt         }n| dk(  rt        }n| dk(  sJ t        }|D ]a  }|d   }|d   }|d }nCt        t        |t        |      z        }t        |      }t        |      dkD  sJ d| d	|        |d
   }||||f c y w)N455_rtr7   r9   )r7   r   zNo files for test  using filter r8   )case_table4case_table5case_table5_rtpjointest_data_pathdictr   len)	rt   filtcasesr   r7   r9   r   use_filtr8   s	            r-   _casesr   E  s     #~	C&    -F|
#<E^TDdO-CDHNEu:> @$TF.?@>y/E8W,,-s   BBr   r   c                 :    t        |       D ]  }t        |d d    y )Nr?   )r   r   rt   r   s     r-   	test_loadr   [  s#    w $$r(#$r/   c                 4    t        |       D ]
  }t        |   y N)r   r   r   s     r-   	test_whosr   a  s    w  $ r/   zversion, fmtsr   r   r   c                 \    t        | d       D ]  }|D ]  }t        |d   |d   |         y )N)r   r   rR   )r   r   )rt   fmtsr   fmts       r-   test_round_tripr   h  s=    
 wT* 2 	2C47DGS1	22r/   c                  :   t        j                  d      } d| d<   d| d<   t        |       }d}d|i}d}t               }	 t	        ||      }t        j                  |d	
      }t        |||       |j                          t        j                  |d
      }t        |dd      }|j                          t        j                  |       t        |d   j                         |d   j                         t        |             y # t        j                  |       w xY w)N)   r   gffffff@rR   r?   g      @)r0   r@   	gzip_testrv   r   wb)moder   ry   TFr   r   )r(   zerosr   r   r   gzipopenr   closer   shutilrmtreer
   r   repr)	xdenserv   r7   r9   r   tmpdirfnamer   r   s	            r-   test_gzip_simpler   r  s    XXgFF3KF3K&ADAwHFYF
fT"YYu40

HV4YYu40
dUKffSk113&sm335&*6l4 	fs   A6D Dc                  J   t               } 	 t        t        j                  d            }t	        | d      }t        ||       t        j                  |       t        ||       t        |       t        j                  |       t        |d      }t        ||       |j                  d       |j                          t        |d      }t        |       |j                  d       |j                          t        j                  |        y # t        j                  |        w xY w)N)rR   rR   rv   za.matr   r   ry   )r   r   r(   r   r   r   osunlinkr   r   r   r   r   r   )r   rv   r   fs       r-   test_multiple_openr     s    YF288F#$fg& 	q
		%q
		% 1	q			
	q			ffs   C)D D"c                      t        t        t        d            } t        t	        |       dkD         | D ]5  }t        |d      }t        t        t        |d       |j                          7 y )Nztesthdf5*.matr   ry   Tr   )
r   r   r   r   r   r   assert_raisesNotImplementedErrorr   r   )	filenamesfilenamefps      r-   
test_mat73r     sa    no.0IC	NQ (D!)'+	- 	
r/   c                      t        t        d      } t        j                         5  t        j                  d       t        | d       t        | d       d d d        y # 1 sw Y   y xY w)Nztestdouble_7.1_GLNX86.materrorTr   F)r   r   warningscatch_warningssimplefilterr   )r   s    r-   test_warningsr     sT     ."=>E		 	 	" /g&-./ / /s   0AA'c                      t               } t        | dddiid       t        |       d   }t        |j                  d       t        |j
                  t        j
                  t                     t        |d   d u        y )Ndr4   rR   r   r   rU   rF   	r   r   r   r   r)   r3   r(   rw   r   )siobacks     r-   test_regression_653r     sc     )CC#qeS)3<DU#RXXf-.DIr/   c                     d} d| z  }t        j                  d|t        fg      }t        t	               d|id       d| dz   z  }t        j                  d|t        fg      }t        t        t        t	               d|id       y )	N   rK   rU   r2   
longstructr   r   r4   r(   r   rw   r   r   r   
ValueErrorlimfldnamest1s      r-   test_structname_lenr
    sz    
CCiG
((5'6!2 3
4CGIc*37SUmG
((5'6!2 3
4C*gwy%c3r/   c                      t        j                  ddt        fg      } t        t        t
        t               d| idd       y )NrU   my_fieldnamer2   	my_structr   Tr   long_field_names)r(   r   rw   r   r  r   r   )r  s    r-   (test_4_and_long_field_names_incompatibler    s7    ~f&=%>?I*gwyy)#Nr/   c                     d} d| z  }t        j                  d|t        fg      }t        t	               d|idd       d| d	z   z  }t        j                  d|t        fg      }t        t        t        t	               d|idd       y )
N?   rK   rU   r2   r  r   Tr  r4   r  r  s      r-   test_long_field_namesr    s    
CCiG
((5'6!2 3
4CGIc*3MSUmG
((5'6!2 3
4C*gwy%c4Ir/   c                     d} d| z  }t        j                  dt              }t        j                  d|t        fg      }||d<   ||d<   t	        t               d|id	d
       t        t        t        t               d|id	d       y )Nr  rK   re   r2   rU   rF   rf   r  r   Tr  F)r(   r   rw   r   r   r   r   r  )r  r  rS   r	  s       r-   test_long_field_names_in_structr    s     CCiG::e&)D
((5'6!2 3
4CDIDIGId+CN *gwy&sULr/   c                      t        j                  dt              } d| d<   d| d<   t        t	               d| id	       t        j                  d
t              } d| d<   t        t	               d| id	       y )Nre   r2   HellorF   Worldrf   rv   r   r   rU   zHello, world)r(   r   rw   r   r   )cellss    r-   test_cell_with_one_thing_in_itr    sc     JJu6*EE#JE#JGIU|C0JJu6*EE#JGIU|C0r/   c                  `   t        t                     } t        | j                  g        dg| _        t        | j                  dg       t        | j                  d       d| _        t        | j                  d       t        | j
                  d       d| _        t        | j
                  d       y )NavarFT)r!   r   r   global_varsunicode_stringsr  )mfws    r-   test_writer_propertiesr     s    

#C"%hCO6(+$$e,C$$d+%%u-C%%t,r/   c                  $   t               } t        |       }t        j                  d      }|j	                  d|i       t        | j                               }| j                  d       | j                  d       |j	                  d|i       t        |t        | j                               z
  dkD         | j                  d       | j                  d       |j	                  d|i       t        t        | j                               |z
  dk         y )N
   aaaaar   aaaar0   aaaaaa)
r   r!   r(   r   put_variablesr   getvaluetruncater   r   )r   wtrr,   w_szs       r-   test_use_small_elementr+    s    
)C

C
((2,Cwn%s||~DLLOHHQKvsm$D3s||~&&*+LLOHHQKxo&C$&*+r/   c                  N   t        j                  dggdt        fdt        fg      } t        t        fD ]o  } |ddg      }t               }t        |d|i       |j                  d       t        |      d   }t        |j                  j                  d	       t        ||        q y )
Nre   rK   br2   )rK   r4   )r-  rR   r   r   rK   r-  )r(   r   rw   r   r   r   r   r   r   r   r3   namesr	   )ab_exp	dict_typer   streamvalss        r-   test_save_dictr4  2  s     XXxj#vf(FGFK( )	x*+$Avv&TZZ%%z24()r/   c                     t        j                  d      } dD ]  }t               }t        |d| i|       t	        |      }t        |d   j                  d       t               }t        |d| i|d       t	        |      }t        |d   j                  d       t               }t        |d| i|d	       t	        |      }t        |d   j                  d        y )
Nr@   r   onedr   )r4   r@   column)r   oned_as)r@   r4   row)r(   aranger   r   r   r   r)   )r,   r   r2  r3  s       r-   test_1d_shaper;  A  s    
))A,C 0f5vT&\''0 	" vT&\''/	 vT&\''/'0r/   c                     t        j                  d      j                  d      } d| d<   t               }t	        |d| i       t        |j                               }t        |      }t        |d   |        t               }t	        |d| id       t        |j                               }t        |      }t        |d   |        t        ||kD         | j                         }d|d<   t               }t	        || |d	d
       t        |      }t        |d   |       t               }t	        || |d	d       t        |      }t        |d   |       y )Nd   )r@   r   r4   )rR   r"  r,   T)do_compressionrF   )r,   arr2Fr?  )r(   r   reshaper   r   r   r'  r   r	   r   copy)r,   r2  raw_lenr3  compressed_lenr?  s         r-   test_compressionrD  Z  s   
((3-


'CCIYFFU3K &//#$G6?DtE{C(YFFU3K5*+N6?DtE{C(Gn$%88:DDIYFF3t,UC6?DtF|T*YFF3t,TB6?DtF|T*r/   c                  f    t               } t        | dt        j                  dt              i       y )NAr4   r2   )r   r   r(   r   rw   )r2  s    r-   test_single_objectrG  u  s#    YFFS!6234r/   c                     t        t        d      } t        | d      }t        d|v        t        d|v        t	        | d      \  }}|j                  d      }t        d|v        |j                  j                          y )Nztest_skip_variable.matTr   firstsecond)r   r   r   r   r   get_variablesr   r   )r   r   factoryfile_openeds       r-   test_skip_variablerN  z  sz     ^$<=H 	40AGqLHM .hNG[ 	h'AHMr/   c                     t        t        d      } t        | d      }|d   }t        |j                  d       t        |j
                  t        j
                  t                     t        |d   d u        t               }t        j                  dd	      }t        |d
|i       t        |      }|d
   }t        ||       y )Nztest_empty_struct.matTr   rK   rU   rF   r   Ur2   r,   )r   r   r   r   r)   r3   r(   rw   r   r   r   r   r	   )r   r   rK   r2  r,   a2s         r-   test_empty_structrR    s    ^$;<H 	40A	#A% "((6*+AcFdNYF
((2S
!CFU3K A	
5Br3r/   c                      t               } t        | di i       t        |       }|d   }t        |j                  d       t        |j
                  t        j
                  t                     t        |d   d u        y )Nr,   rU   rF   r   )r2  r   rK   s      r-   test_save_empty_dictrT    s^    YFFUBK A	%A% "((6*+AcFdNr/   c                 d    d}|D ]  }t        j                  | |k(        sd} n t        |       y)zH Assert `output` is equal to at least one element in `alternatives`
    FTN)r(   allr   )outputalternatives	one_equalr9   s       r-   assert_any_equalrZ    s;     I  66&H$%I Ir/   c            	         t        t        j                  dg            } t        | j                  d       t        | d       t        j                  dgddg      }t        j                  dgddg      }||f}t        t        ddd	      |       t        j                  t        d
      5  t        t        dddd      |       d d d        t        t        dddd      |       t        j                  t        d
      5  t        t        dddd      |       d d d         G d d      }|}d|_
        d|_        t        t        |      |       t        g       } t        | j                  d       t        | j                  j                  t        j                         t        d      } t        | j                  d       t        | j                  j                  t        j                         t        t        d       d u        t        t        d      j                  j                  t        j                          t        d      } t        | j                  d       t        | j                  j                  t        j                  d      j                  j                         t        | d       t        t        i       t"        u        t        t        t%                     d u         G d d      }t        t         |             t"        u        t        ddi      d   } t        | j                  d       t        | j                  j                  t        j&                         t        j                  t        d
      5  t        t        ddi      t"        u        d d d        t        j                  t        d
      5  t        t        ddi      t"        u        d d d        t        j                  t        d
      5  t        t        ddd      t        j                  dgdg             d d d        y # 1 sw Y   wxY w# 1 sw Y   .xY w# 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   y xY w)Nr4   )r4   re   )rK   |O8)r-  r\  r2   )rR   r4   rR   r.  Starting field name withmatchr?   )rK   r-  _c)rK   r-  r=  )rK   r-  99c                       e Zd Zy) test_to_writeable.<locals>.klassN__name__
__module____qualname__r   r/   r-   klassrc        r/   rh  r   r   a stringc                       e Zd Zy)test_to_writeable.<locals>.CNrd  r   r/   r-   Crm    ri  r/   rn  rK   1_a)ro  r   rR   )r   r\  )r$   r(   r   r   r)   r	   rZ  pytestwarnsr   rK   r-  r3   r   float64r   r   r%   rw   object_)res	expected1	expected2rX  rh  crn  s          r-   test_to_writeablerz    sW   
rxx}
%CD!sA&,)EFI&,)EFIy)L\qQ-0,?	o-G	H M1!!&<=|LM \qQA"67F	o-G	H M1!!&<=|LM  	A
AC
AC\!_l3
r
CD!,
r
CD!,L$&'j)//44bgg>
q/CB!!2!2!7!78sAL 112L"d*+  L!223
Qx
 
%CD!,	o-G	H <c!W%)::;< 
o-G	H =d1X&*;;<= 
o-G	H 9\qq/24&7	99 9kM M
M MT< <= =9 9s<   -P<Q	QQ"0Q.<Q	QQ"Q+.Q7c                  H   ddg} t        j                  d|       }d|d   d<   d|d   d	<   d
|d   d<   d|d   d	<   t               }t        |d|i       t	        |d      }|d   d   }t        |j                  d       t        |j                  d       t	        |d      }|d   d   }t        |d   d       t        |d	   d       t        |j                  t        j                  ddg             |d   j                  d   }t        |d   d
       t        |d	   d       y )N)f1f8)f2S10rq  r2   g      ?r   r|  pythonr~  c   r4   znot perlr,   Fr   rF   T)r|  O)r~  r  )
r(   r   r   r   r   r   r|  r~  r3   flat)dtr,   r2  r   a20a21s         r-   test_recarrayr    s#   


B
((4r
"CCF4LCF4LCF4LCF4LYFFUCL!/A
E(3-C".A
E(3-CTC TH%BHHk&1&3 4 5
E(--
CTBTJ'r/   c                  Z    G d d      }  |        }d|_         d|_        t               }t        |d|i       t	        |d      }|d   d   }t        |j                   d       t        |j                  d       t	        |d	      }|d   d   }t        |d
   d       t        |d   d       y )Nc                       e Zd Zy)test_save_object.<locals>.CNrd  r   r/   r-   rn  r  "  ri  r/   rn  r4   rk  ry  Fr   rF   Tfield1field2)r  r  r   r   r   r   )rn  ry  r2  r   c2s        r-   test_save_objectr  !  s     	AAHAHYFFS!H/A	
3BAJ'.A	
3BHq!Hz*r/   c                     t        j                  d      j                  dd      } t               }t	        |d| i       t        |      }|j                         }|d   }t        ||        t        |d      }t        |j                         d   | j                  d             d|_        t        ||        t        |t        j                        }t        |j                         d   |        t        |t        j                        }t        t        |j                         t        j                  |_        t        |j                         d   |        t        j                  d	g      } |j!                  d
       |j#                  d
       t	        |d| i       t        |      }t        |j                         d   |        t        |d      }t        j$                  t        j                  t'        | j)                               d            }t        |j                         d   |       d|_        t        |j                         d   |        y )NrA   r4   rK   T
squeeze_me)rA   F)
byte_orderrk  r   )chars_as_stringsU1r2   )r(   r:  r@  r   r   r"   rK  r	   r  bocnative_codeswapped_coder   	Exceptionr  r   r(  r   
atleast_2dlistr   r  )r,   r2  rdr	back_dictrarrcarrs         r-   test_read_optsr  3  s    ))A,

q
#CYFFS#J

 C!!#IS>DtS!
D
1Cs((*3/T1BCCNtS!
COO
<Cs((*3/5 C,<,<
=C)S../__CNs((*3/5
((J<
 C
OOA
KKNFS#J

 Cs((*3/5
%
8C==$sxxz"2$?@Ds((*3/6Cs((*3/5r/   c                     t        t        d      } t        | d      }t        |      }|j	                         }|j                          t        |d   t        j                  g d             t               }t        |dt        j                  dg      i       t        |      }|j	                         }t        |d   t        j                  g d             |j                  d       |j                  d       t        |dt        j                  g d      i       t        |      }|j	                         }t        |d   t        j                  g d             |j                          y )Nzsingle_empty_string.matry   rK   r  r2    r   )r   r   r   r"   rK  r   r	   r(   r   r   r   r(  r   )estring_fnamer   r  r   r2  s        r-   test_empty_stringr  V  s   .*CDM	mT	"B

CAHHJqvrxx$78 YFFS"((B4.)*

 CAqvrxx$78
OOA
KKNFS"((2T234

 CAqvrxx$78
LLNr/   c                      dd l } t        df| j                  dffD ]I  \  }}t        t	        t
        |      d      5 }t        |      }t        ||j                         d d d        K y # 1 sw Y   VxY w)Nr   corrupted_zlib_data.matzcorrupted_zlib_checksum.matry   )	zlibr  r   r   r   r   r"   r   rK  )r  excr   r   r  s        r-   test_corrupted_datar  q  sx    "$=>

$ABD 2
U%.5 	2 $C#s001	2 	22	2 	2s   "A((A1	c                      t        t        t        d      d      5 } t        | d      }|j	                          d d d        y # 1 sw Y   y xY w)Nr  ry   F) verify_compressed_data_integrity)r   r   r   r"   rK  )r   r  s     r-   )test_corrupted_data_check_can_be_disabledr  z  sC    	eN$=>	E R%H  s   AAc            	      d   dD ]  } t        t        t        |       d      }t        |      }|j	                         }|j                          t        |d   t        j                  dgdggt                     t        |d   t        j                  dd	gd	d
ggt        j                                y )N)zbig_endian.matzlittle_endian.matry   stringshelloworldr2   floats       @      @      @)r   r   r   r"   rK  r   r	   r(   r   rw   float32)r   r   r  r   s       r-   test_read_both_endianr    s    8 
C%.5R 

1Y<88gY&-Y%07=?	@ 	1X;88b"X&("X%/68jjB	C
Cr/   c            	         t        j                  ddgddgg      } t        j                  d      j                  d      }t        j                  ddgd	      }t	               }t        || j                         j                  | j                  j                               |j                         j                  |j                  j                               |j                         j                  |j                  j                               d
       t        |      }|j                         }t        |d   |        t        |d   |       t        |d   |       |j                          y )Nr  r  r  rA   r   r  r  rP  r2   )r  intsuni_arrr  r  r  )r(   r   r:  r@  r   r   byteswapviewr3   newbyteorderr"   rK  r	   r   )	float_arrint_arrr  r2  r  r   s         r-   test_write_opposite_endianr    s     2r(r($ %Iiil""6*Ghh)5GYFF$$&++IOO,H,H,JK  "''(B(B(DE##%**7==+E+E+GH 
 
 CAq{I.qy'*q|W-
LLNr/   c                  R   t        t        t        d      d      5 } t        | d      }|j	                         }d d d        t        j                  dgdggt
        j                        }t        d   |       t        |d   j                  |j                         y # 1 sw Y   exY w)Nztestbool_8_WIN64.matry   T	mat_dtypeFr2   r}   )r   r   r   r"   rK  r(   r   r   r	   r   r3   )fobjr  r   rv   s       r-   test_logical_arrayr    s     
eN$:;T	B  dTT2  	4&5'""((3Aq~q);%%qww/   s   BB&c                     t               } t        j                  g d      }t        | d|i       | j	                  d       t        |       }|j                          |j                          |j                         \  }}t        |j                  t        j                         t        |j                  d       |j                  |d      }t        |j                  j                   t        j"                         y )N)FTFbarrayr   TF)r   r(   r   r   r   r"   initialize_readread_file_headerread_var_headerr   mclassmio5pmxUINT8_CLASS
is_logicalread_var_arrayr3   r   uint8)r2  barrreaderhdr_vars         r-   test_logical_out_typer    s     YF88()DFXt$%
KKNF#F

##%FCU001&


U
+C*r/   c                      t               } t        | dt        j                  d      i       t	        |       }|d   j
                  dk(  sJ y )Nr   )r"  r   )r   r   r(   emptyr   r)   )r2  r   s     r-   test_roundtrip_zero_dimensionsr    s@    YFFS'*+,AS6<<7"""r/   c                      t               } t        j                  d      j                  d      }t	        t
        t        | d|idd       y )N   )rR   r?   r0   rK   Tr   )r   r(   r:  r@  r   r  r   )r2  r,   s     r-   test_mat4_3dr    s7    YF
))B-


(C*gvSz4Er/   c                  |   t        t        d      } t        | d      }t        |      }|j	                         }|j                          t        |d   t              sJ t               }t        |      }t        j                  t        d      5  t        t        |j                  |       d d d        y # 1 sw Y   y xY w)Nztestfunc_7.4_GLNX86.matry   testfuncr]  r^  )r   r   r   r"   rK  r   r   r   r   r!   rr  rs  r   r   r   r&  )func_egr   r  r   r2  r)  s         r-   test_func_readr    s    N$=>G	gt	B

CAHHJa
m^444YF

 C	o-G	H ;mS%6%6:; ; ;s   B22B;c                     t        t        d      } t        | d      }t        |d      }|j	                         }|j                          t        |d   j                  j                  d       t        | d      }t        |d      }|j	                         }|j                          t        |d   j                  j                  d       y )	Nztestmatrix_6.1_SOL2.matry   Fr  rC   uTr   )	r   r   r   r"   rK  r   r   r3   kind)	double_egr   r  r   s       r-   test_mat_dtyper    s    n&?@I	i	B
u
-CAHHJ<&&++S1	i	B
t
,CAHHJ<&&++S1r/   c                      dt        d      i} t               }t        |d| i       t        |d      }t	        |d   d   d   j                         t        j                  d             y )Nsparsefieldr0   rK   Tr   rF   )r   r   r   r   r	   r   r(   eye)str2  r   s      r-   test_sparse_in_structr    s[     1	&BYFFSH.Aqvd|M2::<bffQiHr/   c                  t    t               } ddddi}t        | |       t        | d       t        | dd       y )	Nr  r4   rR   ri   Fr   T)r   r  )r   r   r   )r2  in_ds     r-   test_mat_struct_squeezer    s8    YF#$DFDFU+FUt<r/   c                     t               } dggddddd}t        | |       t        | d      }t        t	        |d	   t
                     t        t	        |d
   t                     t        t	        |d   t        j                               y )N皙?zmy namer4   rR   ri   )scalarr:   r  Tr  r  r:   r  )	r   r   r   r   r   floatr   r(   r   )r2  r  out_ds      r-   test_scalar_squeezer    sq    YFug9KLDFDFt,EJuX./JuX,-JuT{BJJ/0r/   c                     t               } t        j                  ddg      }t        j                  ddg      }t        | t	        |             t        |       }t        |d   |       | j                  d       | j                  d       |j                  d      }t        j                  t        j                  |j                  d|	      }t        | t	        |             t        |d   |       | j                  d       | j                  d       |j                  d
      }|j                  d
      }t        | d|i       t        |       }t        |d   |       y )Nr  FoobzFoob )rK   rK   r   F)order)r)   r3   r  bufferrP  )r   r(   r   r   r   r   r	   r(  r   tobytesr   rK   r)   r3   astype)r2  in_arrout_arrrv  in_strin_from_strin_arr_u	out_arr_us           r-   test_str_roundr  	  s   YFXXw'(Fhh)*GFD6N#
&/Cs3x)
OOA
KKN^^#^&F**177#)<<#&$*,K FD;'(s3x)
OOA
KKN}}S!Hs#IFS(O$
&/Cs3x+r/   c                      t               } t        | ddddi       t        |       }|d   j                  j                  }t        t        |      ddh       y )NrK   r4   rR   r.  r-  )r   r   r   r3   r/  r   set)r2  rv  field_namess      r-   test_fieldnamesr  &  sK    YFFSqa.)*
&/Cc(..&&K[!C:.r/   c                  x   g d} t        t        d      g ft        t        d      | ffD ]  \  }}t        |      }t        t	        |j                               t	        ddg|z                t        |d      }t        t	        |j                               t	        dg|z                t        |dg      }t        t	        |j                               t	        dg|z                t        |dg      }t        t	        |j                               t	        dg|z                t        |d      }t        t	        |j                               t	        dg|z                t        |g       }t        t	        |j                               t	        |             dg}t        ||      }t        |dg        y )N)__globals__
__header____version__ztestmulti_4.2c_SOL2.matztestmulti_7.4_GLNX86.matrK   rJ   )variable_names)rJ   )r   r   r   r   r  keys)mat5_sys_nameseg_filesys_v_namesvarsvnamess        r-   test_loadmat_varnamesr  /  sv   %N 
~8	92>6B89GAI!J ( wS%sC>K+G'HIws3S%sC5;+>'?@wu5S%sC5;+>'?@wy9S%sG9{+B'CDwz:S%sG9{+B'CDwr2S%s;'78wv6VgY'#(r/   c                  <   t        j                  d      } t               }dD ]x  }|j                  d       |j	                  d       t        |d| j                  |      i       t        |      }t        t        j                  |      |d   j                         z y )Nr"  )r}  f4i8i4i2i1u8u4u2u1c16c8r   r,   )
r(   r:  r   r(  r   r   r  r   r   r3   )r,   r2  dtsr  s       r-   test_round_typesr  H  sz    
))B-CYF0 7A

301vRXXc]DK$5$567r/   c                    	 dt        t        j                  d            fdt        d      fdt        d      ff	 G 	fdd      } t               }t	        | |               t        |      }t        t        |      d       t        d      D ]8  }||   \  }}	|   \  }}t        ||       t        |      }t        ||   |       : y )	Nr,   r"  mystrrk  mynumc                       e Zd Z fdZy) test_varmats_from_mat.<locals>.Cc                     S r   r   self
names_varss    r-   r   z&test_varmats_from_mat.<locals>.C.items]  	    r/   Nre  rf  rg  r   r&  s   r-   rn  r"  \      	r/   rn  r?   )r.   r(   r:  r   r   r#   r   r   ranger   r	   )
rn  r2  varmatsr   r7   
var_streamexp_nameexp_resrv  r&  s
            @r-   test_varmats_from_matr0  U  s    %		"./E*-.E"I&(J
  YFFACv&GWq!1X /"1:j&qM'T8$j!3t9g./r/   c                      t        t        d      } t        | d      }t        |      }|j	                         }|j                          t        |d   j                  d       y )Nzone_by_zero_char.matry   r  rj  )r   r   r   r"   rK  r   r   r)   )r  r   r  r   s       r-   test_one_by_zeror2  k  sN    N$:;G	gt	B

CAHHJ5&r/   c                  ^    t        t        d      } t        |       }t        |d   ddgg       y )Nztest_mat4_le_floats.matrK   r  g333333?)r   r   r   r	   )
mat4_fnamer  s     r-   test_load_mat4_ler5  u  s/     ~'@AJ:DtCyC:,/r/   c                  v    t               } ddi}t        | |d       t        |       }t        |d   |d          y )N
second_catu   Schrödingerr   r   )r   r   r   r   )bior  var_backs      r-   test_unicode_mat4r:  }  s<    
)C
(CCS!s|H,'\):;r/   c            	      B   t        t        d      } t        | dd      }|d   }t        t	        |      xr |j
                  dk(         t        |j                  j                  t        j                         t        |j                         g dg dg dg d	g d	g       y )
Nzlogical_sparse.matTFr   
sp_log_5_4csc)TTTF)FFTF)FFFF)r   r   r   r   r   r   r   r3   r   r(   r   r	   r   )r   r   log_sps      r-   test_logical_sparser?    s     ^$89H 	4%@A|_FHV7%!78""BHH-v~~'13344	67r/   c                     t               } dd l}|j                  j                  ddgddgg      }t	        | t        |             | j                  d       t        | d      }|j                  j                  |d         rJ t        | d      }|j                  j                  |d         sJ t        |       }|j                  j                  |d         sJ t        |d   j                  |j                         t        |d   j                         d       | j                  d       t        |       }|j                          |j                          |j                         \  }}t!        |j"                  d       y )Nr   r   F)r   rv   Tr4   )r   scipy.sparserD   	csr_arrayr   r   r   r   
isspmatrixr	   r)   r   r"   r  r  r  r   nzmax)r   scipyempty_sparserv  r  r  r  s          r-   test_empty_sparserG    s9   
)C<<))Aa5!A-8LC%&HHQK
#
&C||&&s3x000
#
%C<<""3s8,,,
#,C<<""3s8,,,s3x~~|'9'9:s3x'')1- HHQKC F

##%FCAr/   c                  B    t               } t        t        t        |        y r   )r   r   r   r   )r   s    r-   test_empty_mat_errorrI    s    
)C,-r/   c                  
   t        t        d      } t        |       }t        |d   t	        j
                  d      d d d f          t        t        d      } t        t              5  t        |        d d d        y # 1 sw Y   y xY w)Nzmiuint32_for_miint32.matan_arrayr"  zbad_miuint32.mat)r   r   r   r   r(   r:  r   r  r   rv  s     r-   test_miuint32_compromiserM    sl     ^%?@H
(
CZ"))B-a"89^%78H	z	"   s   $A99Bc                      t        t        d      } t        |       }t        |d   dgg       t        t        d      } t	        t
              5  t        |        d d d        y # 1 sw Y   y xY w)Nzmiutf8_array_name.mat
array_namer4   zbad_miutf8_array_name.mat)r   r   r   r   r   r  rL  s     r-   !test_miutf8_for_miint8_compromiserP    s^    ^%<=H
(
C\"aSE*^%@AH	z	"   s   A!!A*c                  x    t        t        d      } t        |       }t        |d   dj	                  dd             y )Nzbroken_utf8.mat
bad_strings    am brokenutf8replace)r   r   r   r   decoderL  s     r-   test_bad_utf8rV    s8    ^$56H
(
C\""))&)<>r/   c                 |    t         j                  j                  t        |       d      }ddddi}t	        ||       y )Ntest.matrK   r4   test_str)r-  ry  )r   pathr   r   r   )r   r   	test_dicts      r-   test_save_unicode_fieldr\    s3    ww||CK4H!
+,IHi r/   c                      G d d      } |       }t         j                  j                  t        |       d      }t	        |d|i       t        |      }t        |d   t        j                  |             y )Nc                       e Zd ZddZy)0test_save_custom_array_type.<locals>.CustomArrayNc                 L    t        j                  d      j                  dd      S )Ng      @rR   r?   )r(   r:  r@  )r%  r3   rA  s      r-   	__array__z:test_save_custom_array_type.<locals>.CustomArray.__array__  s    99S>))!Q//r/   )NN)re  rf  rg  ra  r   r/   r-   CustomArrayr_    s    	0r/   rb  rX  rK   )	r   rZ  r   r   r   r   r	   r(   r   )r   rb  rK   r   outs        r-   test_save_custom_array_typerd    s[    0 0 	Aww||CK4HHsAh
(
Cs3x!-r/   c                  X    t        t        t        d       t        t        t        d       y )NzNotExistentFile00.matNotExistentFile00)r   OSErrorr   r   r/   r-   test_filenotfoundrh    s    '7$;<'7$78r/   c                  *   t        t        d      } t        | d      }t        | d      }t        t	        |d   t
                     t        t	        |d   t        j                               t        |d   d   t        j                  g d             y )Nztestsimplecell.matTsimplify_cellsFsmycell)rK   r-  ry  )
r   r   r   r   r   r   r(   r   r	   r   )r   res1res2s      r-   test_simplify_cellsrp    sm    ^%9:H8D1D8E2DJtCy$'(JtCy"**-.tCy*BHH_,EFr/   zversion, filt, regex))r   z_4*_*N)r4   z_5*_*N)r4   z_6*_*N)r4   _7*_*z^((?!hdf5).)*$)rR   rq  z.*hdf5.*)r4   z8*_*Nc                 
   t        t        d| d      }t        |      }|&|D cg c]  }t        j                  ||      | }}t        |      dkD  sJ d|  d|        |D ]  }t        |      }|d   | k(  rJ  y c c}w )Nztest*z.matr   zNo files for version r   )r   r   r   rer_  r   r   )rt   r   regexr   r   filegot_versions          r-   test_matfile_versionrw    s     ^uTF$%78HNE"'M$288E4+@+LMMu:> >
ytf=>> )%d+1~((() Ns   B B c                      t        t        t        d            } t        | d   t              sJ t        | d   j                         d   j                         d   t              sJ y)z,Test that we can read a MatlabOpaque object.parabola.matparabolar?   N)r   r   r   r   r   r   r   r&   s    r-   test_opaquer|    sY    589Dd:&777d:&++-a0557:LIIIr/   c                  d    t        t        t        d      d      } t        | d   t              sJ y)zETest that we can read a MatlabOpaque object when simplify_cells=True.ry  Trj  rz  N)r   r   r   r   r   r{  s    r-   test_opaque_simplifyr~    s*    58NDd:&777r/   c                      t        t              5  t        j                  j                  j
                  j                   ddd       t        t              5  ddlm}  ddd       y# 1 sw Y   (xY w# 1 sw Y   yxY w)z4Test that access to previous attributes still works.Nr   r   )	r   DeprecationWarningrE  iomatlabmio5_paramsr   scipy.io.matlab.miobaser   r  s    r-   test_deprecationr    sa     
(	) 1##001 
(	) 989 9	1 19 9s   /A&A2&A/2A;c                 *   t         j                  j                  d      }| dz  }|j                  d      }|j                  d      }||g}t        |d|idd       i }t	        ||       t        |d   d   d   |       t        |d   d   d	   |       y )
Ni90  z	lists.mat)r@   r?   )rA   r?   r&   T)r  r>  r   r4   )r(   randomdefault_rngr   r   r   )tmp_pathrngoutfile	array_one	array_twolist_of_arraysnew_dicts          r-   test_gh_17992r  (  s    
))


&C$G

5!I

5!I+NG^$!!
 HGHV$Q'*I6HV$Q'*I6r/   c                     t        j                  t        d      t        d      gd      t        j                  ddg      d}| dz  }t        ||d       y )	Nr<   r  r2   r:   )
char_arraystring_arrayztmp.matr   r   )r(   r   r  r   )r  r   r  s      r-   test_gh_19659r  :  sM    hhVd6l;4H(H!56	
A "GGQs#r/   c                      t        t        d      } t        j                  t        j                  k(  rdnd}t        j                  t        |      5  t        |        d d d        y # 1 sw Y   y xY w)Nzdebigged_m4.matz$Not enough bytes to read matrix 'a';z9Variable 'a' has byte length longer than largest possibler^  )	r   r   r(   intpint64rr  r   r  r   )truncated_matr_  s     r-   test_large_m4r  D  s[     .*;<M288# 4H 
 
z	/   s   A&&A/c                      ddl m}  y )Nr   r#   )scipy.io.matlabr#   r  s    r-   test_gh_19223r  T  s    0r/   c                      dt        t        j                  d            fdt        d      ff} t        |        ddddff} t        |        dddd	ff} t        |        y )
N_1r"  r  rk  mymapr4   rR   )rK   _b)rK   1a)r.   r(   r:  check_mat_write_warningr)  s    r-   test_invalid_field_name_warningr  X  si    	uRYYr]#$	%
#$&J J'!1-.0JJ'!1-.0JJ'r/   c                       G  fdd      }t               }t        j                  t        d      5  t	        | |              d d d        y # 1 sw Y   y xY w)Nc                       e Zd Z fdZy)"check_mat_write_warning.<locals>.Cc                     S r   r   r$  s    r-   r   z(check_mat_write_warning.<locals>.C.itemsg  r'  r/   Nr(  r)  s   r-   rn  r  f  r*  r/   rn  r]  r^  )r   rr  rs  r   r   )r&  rn  r2  s   `  r-   r  r  e  sF      YF	o-G	H   s   AAc                  (   dD ]=  } t        j                  t        d      5  t        t	        d| z               d d d        ? t        j                  t        d      5  t        t	        d             d d d        y # 1 sw Y   xY w# 1 sw Y   y xY w)N)rR   r0   r"     z Mat file appears to be truncatedr^      zMat file appears to be corrupts                       )rr  r   r   r   r   ns    r-   test_corrupt_filesr  o  s     *]]<!CE 	*GGaK()	* 	** 
|=
? '%&' '	* 	*' 's   A<B<B	B)ztest%(name)s_*.mat)r   collectionsr   os.pathr   r   r   r   r  r   rs  tempfiler   r   r   r   numpy.testingr	   r
   r   r   r   r   rr  r   r   numpyr(   r   rA  r   r   r   r   scipy.iorE  r  r   r   r   scipy.io.matlab._byteordercodesr  _byteordercodesr  scipy.io.matlab._miobaser   r   r   r   r   scipy.io.matlab._mior   r   r   r    scipy.io.matlab._mio5r!   r"   r#   r$   r%   scipy.io.matlab._mio5_params_mio5_paramsr  __file__r   markthread_unsafe
pytestmarkr.   pir:  r  r@  rJ   r   appendcossinr   rF  r  r+  r  r=   Brw   CAr   CAEr  objarr	transposesqrtexp
st_sub_arrr3   r	  CNr  st2rK   ODTMOm0r   fp_u_strreadrU  u_strr   r   repeatr   r   r   r   r   r   parametrizer   r   r   r   r   r   r   r   r
  r  r  r  r  r   r+  r4  r;  rD  rG  rN  rR  rT  rZ  rz  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r0  r2  r5  r:  r?  rG  rI  rM  rP  rV  r\  rd  rh  rp  rw  r|  r~  r  r  r  r  r  r  r  r  r  s   0r-   <module>r     s   	 # *   	    Q Q  *   B B  F F - -H H N N  - , wx(&1[[&&
 	a			!%(00155x(&
   v&FGHJ   )FBFF5M1A!AB
 BHHUOE!AJ!	eAaj	!A#   x("
   x(	!-
 HHW #"   $h/%y|4
   "2 q)
   h'uRy)
   'u. 
qc
!ugwi	 "(	) *12	 
 GHJ3 V$b!#$ 	!H	!H	"I	"I	!H
 
 %WQr]  
%f	%Ahs   !6*"F+
    &)!3')*   "F+#U&! "   *LBLLDq,=w!GHJ GBGGAJvrvvay/0881=
LM!VMbhhue!= >?M 3 $M 3 %0N C    x($
 RXXe6"
(3
"((5
'3Qbhh'3qcU"((+3rxxV,3!288,3S !288,3S    ( "%
 bhhu.AQa[AB(C"((5'6):(;<C %zl 3C W    !8,"C(
 BHHU?1Av;?@a#ua#uzl##uzl##u    (+!1%
 bhh34F 4 5 ("((5,h7WC5\6
234;C5\6
a9a9a9   x(#
 n&9:DAw'     'ug/
   x(	!-
   $h/%y|4
   i($%)*, Q   "H-#YRYYr1%5%=%=a%BCEF (JV8"3-, J/$ 0$
 J/  0  3*cUO+ 2	248:
/		3N	IL"1-,()02+65
< &B9J(6+$ 6F62C*0+"#F;2I=1,:/(2
7/,'0<7*6.	>!.9G / 2 	)	)J897$$ 1
('G' 	N. B @4s   ee	
e
e