
    d2a                         d dl Zd dlZd dlmc mZ d dlZd dlm	Z	m
Z
mZmZ d dlmZ  ej                    d             Z G d d          ZdS )    N)	DataFrameIndex
MultiIndexSeriesc                  L      G  fddt                        dg di          S )Nc                   *    e Zd Ze fd            ZdS )2gpd_style_subclass_df.<locals>.SubclassedDataFramec                     S N )selfSubclassedDataFrames    \/var/www/html/t/fyr/venv311/lib/python3.11/site-packages/pandas/tests/frame/test_subclass.py_constructorz?gpd_style_subclass_df.<locals>.SubclassedDataFrame._constructor   s	    &&    N)__name__
__module____qualname__propertyr   r   s   r   r   r	      s8        		' 	' 	' 	' 
	' 	' 	'r   r   a         )r   r   s   @r   gpd_style_subclass_dfr      sS    ' ' ' ' ' ' 'i ' ' '
 YYY/000r   c                   .   e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
 Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zej                            dddg          d             Zd Zd Zej        d             Z d Z!d Z"d Z#d  Z$d! Z%d" Z&d# Z'd$S )%TestDataFrameSubclassingc                   	  G 	fddt                     	 G 	fddt                    t          d          t          d          d} |          }t          |          sJ |j        }t          |	          sJ |                                dk    sJ |dd	         }t          |          sJ |                                dk    sJ t          j        d
dg          } ddgddgg|          }t          |d                   sJ t          j        ddg          } ddgddgg|          }t          |d         	          sJ d S )Nc                   0    e Zd Ze fd            Zd ZdS )QTestDataFrameSubclassing.test_frame_subclassing_and_slicing.<locals>.CustomSeriesc                     S r   r   )r   CustomSeriess    r   r   z^TestDataFrameSubclassing.test_frame_subclassing_and_slicing.<locals>.CustomSeries._constructor    s	    ##r   c                     dS NOKr   r   s    r   custom_series_functionzhTestDataFrameSubclassing.test_frame_subclassing_and_slicing.<locals>.CustomSeries.custom_series_function$       tr   N)r   r   r   r   r   r(   )r#   s   r   r#   r!      sG        $ $ $ $ X$    r   r#   c                   J     e Zd ZdZd fdZefd            ZZd Z xZ	S )TTestDataFrameSubclassing.test_frame_subclassing_and_slicing.<locals>.CustomDataFramez
            Subclasses pandas DF, fills DF with simulation results, adds some
            custom plotting functions.
            returnNc                 :     t                      j        |i | d S r   )super__init__)r   argskw	__class__s      r   r/   z]TestDataFrameSubclassing.test_frame_subclassing_and_slicing.<locals>.CustomDataFrame.__init__-   s%      $-"-----r   c                     S r   r   )r   CustomDataFrames    r   r   zaTestDataFrameSubclassing.test_frame_subclassing_and_slicing.<locals>.CustomDataFrame._constructor0   s	    &&r   c                     dS r%   r   r'   s    r   custom_frame_functionzjTestDataFrameSubclassing.test_frame_subclassing_and_slicing.<locals>.CustomDataFrame.custom_frame_function6   r)   r   )r,   N)
r   r   r   __doc__r/   r   r   _constructor_slicedr6   __classcell__)r2   r4   r#   s   @r   r4   r+   '   s         
. . . . . . ' ' ' ' X' #/      r   r4   
   )col1col2r&   r      )Ar>   )r>   Br   r   r   columnsr>   )r>    )r?   rB   )	r   r   range
isinstancer;   r(   r6   r   from_tuples)
r   datacdf
cdf_seriescdf_rowsmcol	cdf_multi
cdf_multi2r4   r#   s
           @@r   "test_frame_subclassing_and_slicingz;TestDataFrameSubclassing.test_frame_subclassing_and_slicing   s   	 	 	 	 	 	 	6 	 	 		 	 	 	 	 	 	 	i 	 	 	$ b		59955od## #///// X
*l333330022d:::: qs8(O44444--//47777 %z:&>??#OaVaV$4dCCC	)C./:::::%y)&<==$_q!fq!f%5tDDD
*S/<8888888r   c                    t          j        g dg ddg d          }d|_        |j        dk    sJ |dg         j        dk    sJ |j        ddgd d f         j        dk    sJ |j        d	d
gd d f         j        dk    sJ |j        d	d
d d f         j        dk    sJ t          j        |          }t          j        ||           |j        |j        k    sJ |j        |j        k    sJ d S )Nr   )XYr   bcindexXXXrO   r   rR   r   r   )tmr   testattrlocilocround_trip_pickleassert_frame_equal	_metadata)r   df	unpickleds      r   test_dataframe_metadataz0TestDataFrameSubclassing.test_dataframe_metadataR   s5   #))))),,OOO
 
 
 {e####3%y!U****vsCj!!!m$-6666w1vqqqy!*e3333 wqsAAAv'50000 (,,	
b),,,|y22222{i0000000r   c                 *   t          j        g dg dg ddg d          }|j        d d df         }t          j        g dt	          d          d	          }t          j        ||           t          |t           j                  sJ |j        d d d
f         }t          j        g dt	          d          d	          }t          j        ||           t          |t           j                  sJ |j        d d df         }t          j        g dt	          d          d	          }t          j        ||           t          |t           j                  sJ |j        dd d f         }t          j        g dt	          d          d	          }t          j        ||           t          |t           j                  sJ |j        d
d d f         }t          j        g dt	          d          d	          }t          j        ||           t          |t           j                  sJ |j        dd d f         }t          j        g dt	          d          d	          }t          j        ||           t          |t           j                  sJ d S )Nr      r=            	   rO   rP   ZrQ   rT   rO   abcrU   namer   rP   rj   r   )r   rc   rf   XYZ)r   r=   rg   rR   rS   )r   rd   rh   )rW   r   rY   SubclassedSerieslistassert_series_equalrD   rZ   r   r^   resexps       r   test_indexing_slicedz-TestDataFrameSubclassing.test_indexing_slicedf   s   #))))))))<<OOO
 
 
 fQQQVn!)))4;;SIII
sC(((#r233333gaaadm!)))4;;SIII
sC(((#r233333fQQQVn!)))4;;SIII
sC(((#r233333fS!!!Vn!)))4;;SIII
sC(((#r233333gadm!)))4;;SIII
sC(((#r233333fS!!!Vn!)))4;;SIII
sC(((#r23333333r   c                      G d dt                     }t          j        t          d          5   |            j         d d d            d S # 1 swxY w Y   d S )Nc                   $    e Zd Zed             ZdS )FTestDataFrameSubclassing.test_subclass_attr_err_propagation.<locals>.Ac                     | j         S r   )i_dont_existr'   s    r   nonexistencezSTestDataFrameSubclassing.test_subclass_attr_err_propagation.<locals>.A.nonexistence   s    ((r   N)r   r   r   r   r{   r   r   r   r>   rx      s-        ) ) X) ) )r   r>   z.*i_dont_exist.*)match)r   pytestraisesAttributeErrorr{   )r   r>   s     r   "test_subclass_attr_err_propagationz;TestDataFrameSubclassing.test_subclass_attr_err_propagation   s    	) 	) 	) 	) 	)	 	) 	) 	)
 ]>1CDDD 	 	ACC	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	s   AAAc                 "   t          j        g dg ddt          d                    }t          j        g dg ddt          d                    }|                    |d	          \  }}t          j        d
t          j        dt          j        dgd
t          j        dt          j        dgdt          d                    }t          j        d
dt          j        dt          j        gd
dt          j        dt          j        gdt          d                    }t          |t           j                  sJ t          j        ||           t          |t           j                  sJ t          j        ||           |j                            |j	                  \  }}t          |t           j
                  sJ t          j        ||j                   t          |t           j
                  sJ t          j        ||j	                   d S )Nr   r   r=   r   rR   ACErT   r   r   rc   )rS   dABDr   axisr   r   r=   ABCDEr   rc   )rW   r   rp   alignnpnanrD   r\   r   rS   ro   rq   )r   df1df2res1res2exp1exp2s          r   test_subclass_alignz,TestDataFrameSubclassing.test_subclass_align   s   $))))),,DKK
 
 
 $))))),,DKK
 
 
 YYsY++
d%bfa+1bfa2KLLw--
 
 
 %aBF+1aBF2KLLw--
 
 
 $ 677777
dD)))$ 677777
dD)))U[[''
d$ 344444
tTV,,,$ 344444
tTV,,,,,r   c                    t          j        g dg ddt          d                    }t          j        g dt          d          d          }|                    |d	
          \  }}t          j        dt
          j        dt
          j        dgdt
          j        dt
          j        dgdt          d                    }t          j        ddt
          j        dt
          j        gt          d          d          }t          |t           j                  sJ t          j        ||           t          |t           j                  sJ t          j	        ||           |                    |          \  }}t          |t           j                  sJ t          j	        ||           t          |t           j                  sJ t          j        ||           d S )Nr   r   r   rT   r   r   xrl   r   r   r   r   r=   r   r   rc   )
rW   r   rp   ro   r   r   r   rD   r\   rq   )r   r^   sr   r   r   r   s          r    test_subclass_align_combinationsz9TestDataFrameSubclassing.test_subclass_align_combinations   s   #))))))$D$DDQVKKXXX			e3GGG XXaaX((
d%bfa+1bfa2KLLw--
 
 

 "261bf%T']]
 
 
 $ 677777
dD)))$ 344444
tT*** WWR[[
d$ 344444
tT***$ 677777
dD)))))r   c                     t          j        ddgi          }|                                D ]A\  }}t          |t           j                  sJ t          j        ||j        |                    Bd S )Nr   r   )rW   r   iterrowsrD   ro   rq   rY   )r   r^   irows       r   test_subclass_iterrowsz/TestDataFrameSubclassing.test_subclass_iterrows   sr    #S1#J//kkmm 	3 	3FAsc2#677777"3q	2222	3 	3r   c                    t          j        g dg dg dgg dg d          }|                                }t          j        g dt	          d          t	          d	          g
          }t          j        ||           d S )Nr   rb   re   rQ   ri   rU   rA   )	r   r   r   rc   r=   rd   rf   rg   rh   	aaabbbccc	XYZXYZXYZrT   )rW   r   stackro   rp   rq   rr   s       r   test_subclass_stackz,TestDataFrameSubclassing.test_subclass_stack   s    #YY			999-!//#OO
 
 
 hhjj!'''[0A0A4CTCT/U
 
 
 	sC(((((r   c                    t          j        g dg dg dg dgt          j        t	          t          t	          d          t	          d                              ddg	          t          j        t	          t          t	          d
          t	          d                              ddg	                    }t          j        ddgddgddgddgddgddgddgddggt          j        t	          t          t	          d          t	          d           t	          d!                              g d"	          t          d#d$gd%                    }|                                }t          j        ||           |                    d          }t          j        ||           t          j        ddgddgddgddgddgddgddgddggt          j        t	          t          t	          d          t	          d           t	          d&                              g d'	          t          d(d)gd%                    }|                    d          }t          j        ||           d S )*Nr:                                  !   (   )   *   +   AABBcdcdaaacccnamesWWXXyzyzwwwyyyr   r:   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   AAAABBBBccddccddyzyzyzyzr   r   r   WrO   rm   WXWXWXWXr   r   r   yz	rW   r   r   rE   rp   zipr   r   r\   r   r^   rt   rs   s       r   test_subclass_stack_multiz2TestDataFrameSubclassing.test_subclass_stack_multi   s   #///1A1A1ACSCSCST(SftF||4455eU^   *SftF||4455eU^  
 
 
 $RRRRRRRR	 (Sj))4
+;+;T*=M=MNNOO+++   3*5111
 
 
$ hhjj
c3'''hhuoo
c3'''$RRRRRRRR	 (Sj))4
+;+;T*=M=MNNOO+++   3*5111
 
 
$ hhuoo
c3'''''r   c                    t          j        g dg dg dg dgt          j        t	          t          t	          d          t	          d                              ddg	          t          j        t	          t          t	          d
          t	          d                              ddg	                    }t          j        ddgddgddgddgddgddgddgddggt          j        t	          t          t	          d          t	          d           t	          d!                              g d"	          t          d#d$gd%                    }|                                }t          j        ||           |                    d          }t          j        ||           t          j        d&d'gddgd(d)gddgd*d+gddgd,d-gddggt          j        t	          t          t	          d          t	          d           t	          d.                              g d/	          t          d0d1gd%                    }|                    d          }t          j        ||           d S )2Nr:   r         (@      *@r   r         6@      7@r   r         @@     @@r   r         E@     E@r   r   r   r   r   r   r   r   r   r   r:   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   rO   r   g      $@g      &@g      4@      5@g      >@g      ?@g      D@g     D@r   r   r   r   r   r   s       r   test_subclass_stack_multi_mixedz8TestDataFrameSubclassing.test_subclass_stack_multi_mixed  s   #$$$$$$$$$$$$	 (SftF||4455eU^   *SftF||4455eU^  
 
 
 $T
T
T
T
T
T
T
T
	 (Sj))4
+;+;T*=M=MNNOO+++   3*5111
 
 
$ hhjj
c3'''hhuoo
c3'''$tttttttt	 (Sj))4
+;+;T*=M=MNNOO+++   3*5111
 
 
$ hhuoo
c3'''''r   c                    t          j        g dg dg dgg dg d          }|                                }t          j        g dt	          d          t	          d	          g
          }t          j        ||           d S )Nr   rb   re   rQ   ri   r   )	r   rc   rf   r   r=   rg   r   rd   rh   	XXXYYYZZZ	abcabcabcrT   )rW   r   unstackro   rp   rq   rr   s       r   test_subclass_unstackz.TestDataFrameSubclassing.test_subclass_unstackY  s    #YY			999-!//#OO
 
 
 jjll!'''[0A0A4CTCT/U
 
 
 	sC(((((r   c                    t          j        g dg dg dg dgt          j        t	          t          t	          d          t	          d                              ddg	          t          j        t	          t          t	          d
          t	          d                              ddg	                    }t          j        g dg dgt          ddgd          t          j        t	          t          t	          d          t	          d          t	          d                              g d	                    }|                                }t          j        ||           |                    d          }t          j        ||           t          j        g dg dgt          ddgd          t          j        t	          t          t	          d          t	          d          t	          d                              g d	                    }|                    d          }t          j        ||           d S )Nr   r   r   r   r   r   r   r   r   r   r   r   r   r   )r:   r   r   r   r   r   r   r   )r   r   r   r   r   r   r   r   r>   r?   r   WWWWXXXXyyzzyyzzcdcdcdcdr   r   r   )r:   r   r   r   r   r   r   r   )r   r   r   r   r   r   r   r   rS   r   ABABABABr   r   r   	rW   r   r   rE   rp   r   r   r   r\   r   s       r   test_subclass_unstack_multiz4TestDataFrameSubclassing.test_subclass_unstack_multih  sH   #///1A1A1ACSCSCST(SftF||4455eU^   *SftF||4455eU^  
 
 
 $---/O/O/OPc
///*Sj))4
+;+;T*=M=MNNOO+++  
 
 
 jjll
c3'''jj
c3'''$---/O/O/OPc
///*Sj))4
+;+;T*=M=MNNOO+++  
 
 
 jj
c3'''''r   c                    t          j        g dg dg dg dgt          j        t	          t          t	          d          t	          d                              ddg	          t          j        t	          t          t	          d
          t	          d                              ddg	                    }t          j        g dg dgt          ddgd          t          j        t	          t          t	          d          t	          d          t	          d                              g d	                    }|                                }t          j        ||           |                    d          }t          j        ||           t          j        g dg dgt          ddgd          t          j        t	          t          t	          d          t	          d          t	          d                              g d	                    }|                    d          }t          j        ||           d S )Nr   r   r   r   r   r   r   r   r   r   r   r   r   r   )r:   r   r   r   r   r   r   r   )r   r   r   r   r   r   r   r   r>   r?   r   r   r   r   r   )r:   r   r   r   r   r   r   r   )r   r   r   r   r   r   r   r   rS   r   r   r   r   r   s       r   !test_subclass_unstack_multi_mixedz:TestDataFrameSubclassing.test_subclass_unstack_multi_mixed  sT   #$$$$$$$$$$$$	 (SftF||4455eU^   *SftF||4455eU^  
 
 
 $888888 c
///*Sj))4
+;+;T*=M=MNNOO+++  

 

 

 jjll
c3'''jj
c3'''$888888 c
///*Sj))4
+;+;T*=M=MNNOO+++  

 

 

 jj
c3'''''r   c                    t          j        g dg dg dd          }|                    ddd          }t          j        dd	d
ddd	d
dd          }d\  |j        _        |j        _        t          j        ||           d S )N)r>   r?   Cr   r?   r>   )Oner   r   Twor   r   )      ?       @      @r   r   r   )rU   rA   valuesrU   rA   r   r   r   r   r>   r?   r   )r   r   r   )rW   r   pivotrU   rm   rA   r\   )r   r^   pivotedexpecteds       r   test_subclass_pivotz,TestDataFrameSubclassing.test_subclass_pivot  s    #777EEE888 
 
 (()H(MM) s55 s55 
 
 6H2X-2
gx00000r   c                     t          j        ddgddgddgddgd	          }t          j        |d
dg          }t          j        g dg dg dg dgg d          }t          j        ||           d S )NJohnMaryDoeBo      @      @      )firstlastheightweightr  r  )id_varsr   r   r  r  r   r   r  r  r   r   r  r  r   r   r  r  r  r  variablevaluer@   )rW   r   pdmeltr\   )r   cheesemeltedr   s       r   test_subclassed_meltz-TestDataFrameSubclassing.test_subclassed_melt  s    ' &)**	 
 
 '6):;;;)...---...---	 ;::
 
 
 	fh/////r   c                 l   t           j                            d           t           j                            d          }t	          j        ddddddd	dd
dddddddt          t          t          d          |                    d          }|j	        |d<   |
                                |
                                z   g dg dg dg dd}t	          j        |          }|                    ddg          g d         }t          j        |ddgdd          }t	          j        ||           d S )N{   r   r   rR   rS   r   r   r   r   ef      @333333?ffffff?皙	@?皙?)A1970A1980B1970B1980rO   id)r   rR   rS   r   r  r  )r  r  r  r  r  r   )  r&  r&    r'  r'  )r   r   r   r   r   r   )rO   r>   r?   yearr%  r(  )rO   r>   r?   r>   r?   )r   j)r   randomseedrandnrW   r   dictr   rC   rU   tolist	set_indexr  wide_to_longr\   )r   r   r^   exp_datar   
long_frames         r   test_subclassed_wide_to_longz5TestDataFrameSubclassing.test_subclassed_wide_to_long  sR    		sIOOA# SS11 SS11 SS11 SS11#eAhh**++ 
 
 84ahhjj(//////888$$$
 
 )(33%%tVn55oooF_R#stvFFF

j(33333r   c                    d d t          j        g dg dg dg dgg d          }|                    fd	           |                    fd
d           t          j        g dg dg dg dgg d          }|                    fdd          }t          |t           j                  sJ t          j        ||           t          j        g dg dg dg dg          }|                    d d          }t          |t           j                  sJ t          j        ||           |                    d dd          }t          |t           j                  sJ t          j        ||           t          j        g dg dg dg dg          }|                    d d          }t          |t           j                  rJ t          j        ||           d S )Nc                 >    t          | t          j                  sJ d S r   )rD   rW   ro   r   s    r   check_row_subclasszJTestDataFrameSubclassing.test_subclassed_apply.<locals>.check_row_subclass  s!    c2#67777777r   c                 >    | d         dk    r| dxx         dz  cc<   | S )Nr  r  r  g      ?r   r6  s    r   stretchz?TestDataFrameSubclassing.test_subclassed_apply.<locals>.stretch  s,    :(**G#Jr   r
  r  r  r  r  r@   c                      |           S r   r   r   r7  s    r   <lambda>z@TestDataFrameSubclassing.test_subclassed_apply.<locals>.<lambda>#      --a00 r   c                      |           S r   r   r;  s    r   r<  z@TestDataFrameSubclassing.test_subclassed_apply.<locals>.<lambda>$  r=  r   r   r   )r   r   r  r  )r   r   r  g      @c                      |           S r   r   )r   r9  s    r   r<  z@TestDataFrameSubclassing.test_subclassed_apply.<locals>.<lambda>0  s    GGAJJ r   r   c                 .    t          j        g d          S Nr   )rW   ro   r   s    r   r<  z@TestDataFrameSubclassing.test_subclassed_apply.<locals>.<lambda>6  s    B$7			$B$B r   c                 
    g dS rA  r   rB  s    r   r<  z@TestDataFrameSubclassing.test_subclassed_apply.<locals>.<lambda>:  
    III r   expand)r   result_typec                 
    g dS rA  r   rB  s    r   r<  z@TestDataFrameSubclassing.test_subclassed_apply.<locals>.<lambda>@  rD  r   )rW   r   applyrD   r\   ro   rq   )r   r^   r   resultr7  r9  s       @@r   test_subclassed_applyz.TestDataFrameSubclassing.test_subclassed_apply  sn   	8 	8 	8	 	 	
 #...---...---	 ;::
 
 
 	0000111
0000q999)...---...---	 ;::
 
 
 ....Q77&""899999
fh///)999iiiIII*VWWBBKK&""899999
fh///--A8LL&""899999
fh///&			999iii'STT--A66fb&<=====
vx00000r   c                     t          j        g dg dg dd          } t          ||                      }t          |t           j                  sJ d S Nr   rb   re   r   rW   r   getattrrD   ro   )r   all_reductionsr^   rI  s       r   test_subclassed_reductionsz3TestDataFrameSubclassing.test_subclassed_reductionsD  s`     #)))))))))$T$TUU,^,,..&""56666666r   c                    t          j        g ddt          j        dddgg dd          }|                                }t          |t           j                  sJ t          j        g dg d	g d
d          }|                                }t          |t           j                  sJ t          j        g dg dg dg dgt          j        t          t          t          d          t          d                              ddg          t          j        t          t          t          d          t          d                              ddg                    }|                                }t          |t           j                  sJ t          j                    }|                                }t          |t           j                  sJ d S )N)r   MylaLewisr   rR  g      8@r   r      )FTTTF)PersonAgeSingle)r   r   r   )r   r=   rd   )rf   rg   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   )rW   r   r   r   countrD   ro   r   rE   rp   r   r   r^   rI  s      r   test_subclassed_countz.TestDataFrameSubclassing.test_subclassed_countK  s   #CCCbfdB3::: 
 
 &""566666#)))))))))$T$TUU&""566666#///1A1A1ACSCSCST(SftF||4455eU^   *SftF||4455eU^  
 
 
 &""566666#%%&""56666666r   c                     t          j        ddgddgdddg          }|                    ddg          }t          |t           j                  sJ d S )Nr   rc   r   )num_legs	num_wingsfalcondogrT   )rW   r   isinrD   rY  s      r   	test_isinz"TestDataFrameSubclassing.test_isinj  sg    #Qq!f55h=N
 
 
 !Q&""89999999r   c                 (   t          j        g dg dg dd          }|                                }t          |t           j                  sJ t          j                    }|                                }t          |t           j                  sJ d S rL  )rW   r   
duplicatedrD   ro   rY  s      r   test_duplicatedz(TestDataFrameSubclassing.test_duplicatedq  s    #)))))))))$T$TUU&""566666#%%&""56666666r   
idx_methodidxmaxidxminc                     t          j        g dg dg dd          } t          ||                      }t          |t           j                  sJ d S rL  rM  )r   re  r^   rI  s       r   test_idxz!TestDataFrameSubclassing.test_idxz  s^    #)))))))))$T$TUU(Z((**&""56666666r   c                    t          j        g dg dg          }t          j        g d          }|                    |          }t	          |t           j                  sJ t          j        g dg dg          }t          j        g d          }|                    |          }t	          |t           j                  sJ d S )N)r   r   )r   r   r   r   )r   r   r   r   )rW   r   ro   dotrD   )r   r^   r   rI  s       r   test_dotz!TestDataFrameSubclassing.test_dot  s    #^^^\\\$BCC--&""566666#^^^\\\$BCC"<<<00&""89999999r   c                    t          j        g dg dg dd          }|                                }t          |t           j                  sJ |                    d          }t          |t           j                  sJ d S )Nr   rb   re   r   FrT   )rW   r   memory_usagerD   ro   rY  s      r   test_memory_usagez*TestDataFrameSubclassing.test_memory_usage  s    #)))))))))$T$TUU""&""566666u--&""56666666r   c                 d   g d}g d}t          j        t          j                            dd          ||          }t          j        t          j                            dd          |d d         |          }|                    |ddd	          }t          |t           j                  sJ d S )
N)r   rR   rS   r   r  )onetwothreefourr=   rc   r   r   Tkendall)r   dropmethod)rW   r   r   r*  r,  corrwithrD   ro   )r   rU   rA   r   r   correlss         r   test_corrwithz&TestDataFrameSubclassing.test_corrwith  s    )))111$IOOAq!!
 
 
 $IOOAq!!rrG
 
 
 ,,si,HH'B$79999999r   c                 t   d}t          j        d|d          }t          j        t          j        t          j        t          j        gt          j        t          j        t          j        gt          j        t          j        t          j        gd|          }|                    |dd                    }t          |t          j                  sJ |                    |d                   }t          |t          j                  sJ |                    d          }t          |t          j                  sJ d S )	Nr   z1/1/199053s)periodsfreqr   rT   rk  z
1989-12-31)	r  
date_rangerW   r   r   r   asofrD   ro   )r   Nrngr^   rI  s        r   	test_asofz"TestDataFrameSubclassing.test_asof  s   mJ>>>#fbfbf-fbfbf-fbfbf- 
 
 
 
 RSS""&""899999R!!&""566666&&&""56666666r   c                     t          j        g dg dg dd          }|                                }t          |t           j                  sJ d S rL  )rW   r   rg  rD   ro   rY  s      r   test_idxmin_preserves_subclassz7TestDataFrameSubclassing.test_idxmin_preserves_subclass  V     #)))))))))$T$TUU&""56666666r   c                     t          j        g dg dg dd          }|                                }t          |t           j                  sJ d S rL  )rW   r   rf  rD   ro   rY  s      r   test_idxmax_preserves_subclassz7TestDataFrameSubclassing.test_idxmax_preserves_subclass  r  r   c                    t          j        g dg dg dd          }|                                }t          |t           j                  sJ |                                }t          |t	          |                    sJ d S rL  )rW   r   convert_dtypesrD   type)r   r   r^   rI  s       r   &test_convert_dtypes_preserves_subclassz?TestDataFrameSubclassing.test_convert_dtypes_preserves_subclass  s    #)))))))))$T$TUU""$$&""899999&5577&$'<"="=>>>>>>>r   c                     t          j        g dg dg dd          }|                    t          j        t          j        t          j        d          }t          |t           j                  sJ d S rL  )rW   r   astyper   int64int32float64rD   rY  s      r   test_astype_preserves_subclassz7TestDataFrameSubclassing.test_astype_preserves_subclass  sh    #)))))))))$T$TUUrzJJKK&""89999999r   c                     t          dg di          }t          j        dg di          }|                    |          sJ |                    |          sJ d S )Nr   r   )r   rW   r   equals)r   r   r   s      r   test_equals_subclassz-TestDataFrameSubclassing.test_equals_subclass  se     iii())$c999%566zz#zz#r   c                     t          j        dg di          }|                    ddgd          }t          j        dg di          }t          |t           j                  sJ t          j        ||           d S )Nr>   r  r   r   ffill)ry  )r   r   r   )rW   r   replacerD   r\   )r   r^   rI  r   s       r   test_replace_list_methodz1TestDataFrameSubclassing.test_replace_list_method  s    #S)))$455QF733)3			*:;;&""899999
fh/////r   N)(r   r   r   rM   r`   ru   r   r   r   r   r   r   r   r   r   r   r   r  r3  rJ  rP  rZ  ra  rd  r}   markparametrizeri  rn  rq  tdskip_if_no_scipyr|  r  r  r  r  r  r  r  r   r   r   r   r      s0       59 59 59n1 1 1(!4 !4 !4F  - - -:* * *83 3 3) ) )7( 7( 7(r<( <( <(|) ) )%( %( %(N0( 0( 0(d1 1 1.0 0 024 4 4:41 41 41l7 7 77 7 7>: : :7 7 7 [\Hh+?@@7 7 A@7
	: 	: 	:7 7 7 : : :7 7 7*7 7 77 7 7? ? ?: : :  0 0 0 0 0r   r   )numpyr   r}   pandas.util._test_decoratorsutil_test_decoratorsr  pandasr  r   r   r   r   pandas._testing_testingrW   fixturer   r   r   r   r   <module>r     s         ) ) ) ) ) ) ) ) )                      1 1 1H0 H0 H0 H0 H0 H0 H0 H0 H0 H0r   