
    Mh]                     \    S SK rS SKrS SKrS SKJrJr  S SKJr	  S r
S rS rS rS rS rg)	    N)	DataFrame
MultiIndexc                 h    U R                  5       nU R                  n[        R                  " X5        g )N)to_numpyvaluestmassert_numpy_array_equal)idxresultexps      \/var/www/html/env/lib/python3.13/site-packages/pandas/tests/indexes/multi/test_conversion.pytest_to_numpyr      s$    \\^F
**C,    c            
      B   / SQn [         R                  " U 5      nUR                  SS9n[        U 5      n[        R
                  " X#5        UR                  5       nXl        [        R
                  " X#5        / SQn [         R                  " U SS/S9nUR                  SS9n[        U 5      nSS/Ul        [        R
                  " X#5        UR                  5       nXl        [        R
                  " X#5        [         R                  " U 5      nUR                  SSS/S9n[        U 5      nSS/Ul        [        R
                  " X#5        UR                  SS/S9nXl        SS/Ul        [        R
                  " X#5        S	n[        R                  " [        US
9   UR                  SS9  S S S 5        Sn[        R                  " [        US
9   UR                  S/S9  S S S 5        [         R                  " [        S5      [        R                  " SSS9/5      nUR                  SS9n[        [         R"                  " [         R$                  " SSS9S5      [         R&                  " [        R                  " SSS9S5      S.5      n[        R
                  " X#5        UR                  5       nXl        [        R
                  " X#5        UR                  SSS/S9n[        [         R"                  " [         R$                  " SSS9S5      [         R&                  " [        R                  " SSS9S5      S.5      n[        R
                  " X#5        UR                  SS/S9nXl        [        R
                  " X#5        g ! , (       d  f       GN= f! , (       d  f       GN= f)N))   one)r   two)   r   )r   r   Findexfirstsecondnames)r   name)r   z1'name' must be a list / sequence of column names.matchz<'name' should have same length as number of levels on index.   20130101   )periodsint64)dtype)r   r   )r   r   )r   from_tuplesto_framer   r   assert_frame_equalr   columnspytestraises	TypeError
ValueErrorfrom_productrangepd
date_rangenprepeatarangetile)tuplesr   r   expectedmsgs        r   test_to_framer7      s   =F""6*E^^%^(F H&+^^FN&+=F""6'81DEE^^%^(F H*H&+^^FN&+ ""6*E^^%w.A^BF H*H&+^^'8!4^5FN*H&+
=C	y	,G$ 
- IC	z	-WI& 
. ##U1Xr}}ZQR/S$TUE^^%^(Fyy1G4a8wwr}}Z;Q?	
H &+^^FN&+ ^^%w.A^BFYYryy':A>ggbmmJBAF	
H &+^^'8!4^5FN&+E 
-	, 
.	-s   M=N=
N
Nc            	      &   [         R                  " [        R                  " SSSS9/ SQ[        R                  " / SQSS9/ S	Q// S
QS9n [        U R                  5       VVs0 s H  u  pX R                  U   R                  _M      nnn[        [        R                  " SSSS9/ SQ[        R                  " / SQSS9/ S	QS
.5      nU R                  SS9nUR                  R                  5       n[        R                  " XT5        X6:X  d   eg s  snnf )N19910905   z
US/Eastern)r!   tz)r   r   r   r   r   r   )ar<   br=   cr>   T)ordered)xr@   yzr@   rA   )datesr<   r=   r>   r   Fr   )r   from_arraysr.   r/   Categorical	enumerater   levelsr#   r   r%   dtypesto_dictr   r&   )miir   original_dtypesexpected_dfdf	df_dtypess          r   test_to_frame_dtype_fidelityrP   [   s    			MM*aLANN94H*		
 '
B @I?RS?RGAtYYq\///?ROS]]:q\J# >M/		
K 
5	!B		!!#I"*''' Ts   !%Dc                      / SQn [         R                  " / SQ/ SQ/ SQ/U S9nUR                  5       R                  R	                  5       nX :X  d   eg )N)rB   r   r<   )r<   r=   r>   )r@   rA   rB   )qwer   )r   rD   r%   r'   tolist)r5   rJ   r   s      r   $test_to_frame_resulting_column_orderrV   w   sK    H				/?;8
B [[]""))+Fr   c                  8   SS/n SS/n[         R                  " XS9n[        R                  " [        SS9   UR                  5         S S S 5        UR                  SS9n[        XUS	9n[        R                  " X45        S S
/n[         R                  " XS9n[        R                  " [        SS9   UR                  5         S S S 5        UR                  SS9n[        XS
S
/S	9n[        R                  " X45        g ! , (       d  f       N= f! , (       d  f       NQ= f)N)r   r   )r       r<   r   z%Cannot create duplicate column labelsr   T)allow_duplicates)r   r'   r   )	r   r$   r(   r)   r+   r%   r   r   r&   )datar   r   r   r5   s        r   test_to_frame_duplicate_labelsr[      s    FD#JE""45E	z)P	Q 
R ^^T^2FE:H&+1IE""45E	z)P	Q 
R ^^T^2FQF;H&+ 
R	Q 
R	Qs   C:/D:
D
Dc                 z    [         R                  " SSS9nU R                  5       n[        R                  " X!5        g )N))foor   )r]   r   )barr   )bazr   )quxr   )r`   r   F)tupleize_cols)r.   Indexto_flat_indexr   assert_index_equal)r
   r5   r   s      r   test_to_flat_indexre      s8    xx	
 
H  F&+r   )numpyr0   r(   pandasr.   r   r   pandas._testing_testingr   r   r7   rP   rV   r[   re    r   r   <module>rk      s8       -F,R(8,,,r   