
    e!h                        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/m0Z0 d dl1m2Z2m3Z3m4Z4m5Z5 d dl6m7Z7m8Z8m9Z9m:Z:m;Z; d dl<mc m)c m=Z> d dl?m@Z@  e eeA      d      ZBej                  j                  ZEd ZFej                  dz   ej                  deI      j                  dd      z  ZKdddideKidgZLeLj                  dddid edg      id       eLj                  dddid ej                  eK      d ej                  eK      z  z   id        ej                  d       ZQ eR eSdd!            eQd <    eR eSdd            eQddd f<   eLj                  d"d#did#eQid       eLj                  d$d%d$id% e eQ      id       eQj                  eU      ZVeVd&xx   dz  cc<   eLj                  d'd(d$id( e eV      id       eLj                  d)ddd*eKeQd*d       eLj                  d+d,did, eFd-      id       eLj                  d.d/did/ ed0g      id        eFg  eFdg       eFdd1gg       eFg d2g      feW      j                  dd-      ZX ed3g      eXd&<   d4d5d4id5eXidgZY eF eFd       eFd1       eFg        eFg        eFd6      feW      j                  dd-      ZZ ej                  d7eW      Z\ eFd      e\d&<   eYj                  d8d9d4id9e\id       eYj                  d:d;d4id;eZid       eYj                  d<d=did= eg d>      id       eYj                  d?d@did@ ej                   ej                   eR eSddA            dB            id        e ej                  d1       ej                  d      ej                  g      j                  dd6      Z`dCD  cg c]  } | eWf c} Za ej                  d7ea      Zb edDg      ebdE   d&<   e`ebdF   d&<   e`dGz  ebdH   d&<   eYj                  dIdJdIidJebid        ej                  dKeW      Zc eFd      ecd&<    ej                  dLeW      ecdM<    eFd1ej                        ecdM   d&<    eFd6ggej                        ecdM   dM<    ej                  dKeW      ecdM   dN<    eFdej                        ecdM   dN   d&<    eFdOej                        ecdM   dN   dM<   eYj                  dPdQd4idQecid        ej                  d7dRD  cg c]  } | eWf c}       Ze eFd      eed&   dS<    ej                  d7dTeWfg      eed&   dU<    edVg      eed&   dU   d&   dT<   eYj                  dWdXdIidXeeid        ej                  dKdRD  cg c]  } | eWf c}       Zf eFd      efd&   dS<    eFd1      efd&   dU<    edYg      efdM   dS<    edZg      efdM   dU<   eYj                  d[d\dIid\efid        ej                  d]D  cg c]  } | eWf c}       Zg e' ej                  d7eg      d^      Zhehd&   Zi ed_g      eid`<    edag      eidb<    ed_g      eidc<    eFd       eidd<    eFd      eide<    eFd      eidf<   eYj                  dgdhdgidhehid        ej eeBdi      dj      Zkekj                         j                  dk      Znekj                          eYj                  dldmdidm eeng      id       eYj                  d$d%d$id% e eQ      id       eYj                  d'd(d$id( e eV      id       eYj                  dndodpido edqgdrgg      id       eYdd Zpepj                  dsdtdgidt ej                  ehd1      j                  dd1      id       du Zrdv Zsdw Ztdx Zudy ZvddzZwej                  j                  dfd{      d|        Zyej                  j                  dfd{      d}        Zzej                  j                  d~dddgfddgfg      d        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 Zd Zej                  j                  dg d      d        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)mat_reader_factoryloadmatsavematwhosmat)MatFile5WriterMatFile5Readervarmats_from_matto_writeableEmptyStructMarker)VisibleDeprecationWarningdatac                  P    t        j                  | i |}t        |      |_        |S )z;Convenience function to return matlab-compatible 2-D array.)npr   r   shape)argskwargsarrs      f/var/www/html/diagnosisapp-backend/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	| |fz  }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
%s, [%d], 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&%3K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   F  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   b  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   i  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   s  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  3  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;  B  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  [  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  v  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  {  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        t        ddd
d      |       t        t        ddd
d      |       t        t        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        t        ddi      t        u        t        t        ddi      t        u        t        t        ddd      t        j                  dgdg             y )Nr4   )r4   re   )rK   |O8)r-  r\  r2   )rR   r4   rR   r.  r?   )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-   klassr`        r/   re  r   r   a stringc                       e Zd Zy)test_to_writeable.<locals>.CNra  r   r/   r-   Crj    rf  r/   rk  rK   1_a)rl  r   rR   )r   r\  )r#   r(   r   r   r)   r	   rZ  rK   r-  r3   r   float64r   r   r$   rw   object_)res	expected1	expected2rX  re  crk  s          r-   test_to_writeableru    s   
rxx}
%CD!sA&,)EFI&,)EFIy)L\qQ-0,?\qQQ"78,G\qQA"67F\qQQ"78,G  	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!,L#a!%667L$q"&77811o.4&79r/   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S10rn  r2   g      ?r   rw  pythonry  c   r4   znot perlr,   Fr   rF   T)rw  O)ry  r}  )
r(   r   r   r   r   r   rw  ry  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>.CNra  r   r/   r-   rk  r    rf  r/   rk  r4   rh  rt  Fr   rF   Tfield1field2)r  r  r   r   r   r   )rk  rt  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_orderrh  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  .  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  Q  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  l  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  u  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        t        |j                  |       y )Nztestfunc_7.4_GLNX86.matry   testfunc)r   r   r   r!   rK  r   r   r   r   r    r   r   r&  )func_egr   r  r   r2  r)  s         r-   test_func_readr    ss    N$=>G	gt	B

CAHHJa
m^444YF

 C-!2!2A6r/   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_arrrq  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  rq  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"  )rx  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  A  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"  mystrrh  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.itemsV  s	    r/   N)rb  rc  rd  r   )r   s   r-   rk  r  U  s    	r/   rk  r?   )r.   r(   r:  r   r   r"   r   r   ranger   r	   )
rk  r2  varmatsr   r7   
var_streamexp_nameexp_resrq  r   s
            @r-   test_varmats_from_matr&  N  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  rg  )r   r   r   r!   rK  r   r   r)   )r  r   r  r   s       r-   test_one_by_zeror(  d  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_ler+  n  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_mat4r0  v  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_sparser5    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_sparserq  r  r  r  s          r-   test_empty_sparser=    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_errorr?    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   rq  s     r-   test_miuint32_compromiserC    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  rB  s     r-   !test_miutf8_for_miint8_compromiserF    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   decoderB  s     r-   test_bad_utf8rL    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-  rt  )r   pathr   r   r   )r   r   	test_dicts      r-   test_save_unicode_fieldrR    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)rb  rc  rd  rW  r   r/   r-   CustomArrayrU    s    	0r/   rX  rN  rK   )	r   rP  r   r   r   r   r	   r(   r   )r   rX  rK   r   outs        r-   test_save_custom_array_typerZ    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_filenotfoundr^    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-  rt  )
r   r   r   r   r   r   r(   r   r	   r   )r   res1res2s      r-   test_simplify_cellsrf    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   rg  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   rematchr   r   )rt   r   regexr   r   filegot_versions          r-   test_matfile_versionrn    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_opaquers  	  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.rp  Tr`  rq  N)r   r   r   r   r   rr  s    r-   test_opaque_simplifyru    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   DeprecationWarningr;  iomatlabmio5_paramsr   scipy.io.matlab.miobaser   rw  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         j                  j	                         5 }|j                  t               t        |d|idd       d d d        i }t        ||       t        |d   d   d   |       t        |d   d   d	   |       y # 1 sw Y   BxY w)
Ni90  z	lists.mat)r@   r?   )rA   r?   r&   T)r  r>  r   r4   )
r(   randomdefault_rngtestingsuppress_warningsfilterr%   r   r   r   )tmp_pathrngoutfile	array_one	array_twolist_of_arrayssupnew_dicts           r-   test_gh_17992r  !  s    
))


&C$G

5!I

5!I+N		%	%	' %3

,-(!%#	%% HGHV$Q'*I6HV$Q'*I6% %s   )'CC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  6  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 possiblerj  )	r   r   r(   intpint64pytestr   r  r   )truncated_matrj  s     r-   test_large_m4r  @  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  P  s    0r/   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                       )r  r   r   r   r   ns    r-   test_corrupt_filesr  S  s     *]]<!CE 	*GGaK()	* 	** 
|=
? '%&' '	* 	*' 's   A<B<B	B)ztest%(name)s_*.mat)r   collectionsr   os.pathr   r   r   r   ry  r   ri  tempfiler   r   r   r   numpy.testingr	   r
   r   r   r   r   r  r   r   numpyr(   r   r7  r   r   r   r   scipy.ior;  r  r   r   r   scipy.io.matlab._byteordercodesrz  _byteordercodesr  scipy.io.matlab._miobaser   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  scipy._lib._utilr%   __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readrK  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  ru  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r
  r  r&  r(  r+  r0  r5  r=  r?  rC  rF  rL  rR  rZ  r^  rf  rn  rs  ru  r}  r  r  r  r  r  r  s   0r-   <module>r     s   	 # *   	    Q Q  *   B B  F F - -; ; N N  - , 6 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
< &<9~(6+$ 6F62C*0+"#F	72I=1,:/(2
7/,'0<7*6.	>!.9G / 2 	)	)J897*$ 1'M& 	N. B @4s   ee
e
e