
    bMh              
          d dl m Z  d dlZd dlZd dlmZmZmZ d dlm	Z
 d dlmZ ej                            d          Zej                            d e
j        d ej        d d          i           e
j         ej        d d          d	          g          d
             Zd Zej                            dee
j        g          d             ZdS )    )datetimeN)	DataFrameIndexSeries)get_groupby_method_argszMignore:Passing a BlockManager|Passing a SingleBlockManager:DeprecationWarningobjA
   namec                    t          | t                    r|dv rt          j        d|            |                     t          j        dd                    }t          |                    d          t          |                     sJ t          ||           }|dk    rt          nd }t          |          j         d}t          j        ||d          5   t          ||          | }d d d            n# 1 swxY w Y   t          j        ||d          5   |j        |g|R  }d d d            n# 1 swxY w Y   h d	}t          | t                     r!||v rt          |t          j                  sJ nt          |t          |                     sJ t          |t                     rt          j        ||           d S t          j        ||           d S )
N>   corrwithzNot applicable for Series and r   r
   fillnaz.fillna is deprecatedF)matchraise_on_extra_warnings>   sizengroupcumcount)
isinstancer   pytestskipgroupbynparange	get_grouptyper   FutureWarning__name__tmassert_produces_warninggetattraggr   SubclassedSeriesassert_frame_equalassert_series_equal)	r   groupby_funcgroupedargswarnmsgresult1result2slicess	            j/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/groupby/test_groupby_subclass.pytest_groupby_preserves_subclassr/      sl    #v E<<#?#?C\CCDDDkk")Ar**++G g''**DII66666"<55D(H44==$D']]#
:
:
:C		#DU	S	S	S 8 80''<00$78 8 8 8 8 8 8 8 8 8 8 8 8 8 8		#DU	S	S	S 3 3'+l2T2223 3 3 3 3 3 3 3 3 3 3 3 3 3 3 ,++F#y!! .lf&<&<'2#6777777'499----- '9%% 1
gw/////
w00000s$   C88C<?C<D55D9<D9c                  N   t          j        g dg dg dd          } d| j        v sJ d| _        |                     d          D ]\  }}|j        dk    sJ d }d	}t          j        t          |d
d
          5  |                     d                              |          }d d d            n# 1 swxY w Y   t          j        dgdz  t          g dd                    }t          j
        ||           |                     d                              |d
          }t          j
        ||           |                     d          ddg                             |          }t          j
        ||           d }t          j        g d          }d|_        |                    | d                                       |          }t          j
        ||           |                    | d                                       |          }t          j
        ||           d S )N)         )r1   r1   r2   )      	   )abctestattrhellor9   c                     t          | t          j                  sJ t          | d          sJ | j        dk    sJ | j        S )Nr:   r;   )r   r   SubclassedDataFramehasattrr:   groups    r.   funcz-test_groupby_preserves_metadata.<locals>.funcE   sL    %!788888uj)))))~((((~    z7DataFrameGroupBy.apply operated on the grouping columnsFr   r   check_stacklevelr3   r   )index)include_groupsr7   r8   c                 l    t          | t          j                  sJ t          | d          sJ | j        S )Nr:   )r   r   r#   r>   r:   r?   s    r.   func2z.test_groupby_preserves_metadata.<locals>.func2]   s9    %!455555uj)))))~rB   )r   r=   	_metadatar:   r   r    r   applyr#   r   r%   r"   )		custom_df_group_dfrA   r*   resultexpectedrH   custom_seriess	            r.   test_groupby_preserves_metadatarQ   <   s   &YYYYYYYYY'W'WXXI,,,,, I ((-- , ,8 G+++++   DC		# %	
 
 
 4 4 ""3''--d334 4 4 4 4 4 4 4 4 4 4 4 4 4 4 "G9q=iiic8R8R8RSSSH68,,,s##))$u)EEF68,,, s##S#J/55d;;F68,,,  
 '			22M$M""9S>2288??F68,,,""9S>2266u==F68,,,,,s   ;)B00B47B4c                 |    | t          d                                t                    g dt          ddddd          t          ddddd	          t          dd
ddd          t          dd
dd
d          t          ddddd          t          ddddd          gd          }|                    d          }d}t          j        t          |dd          5  |                    d          	                    d          
                                }d d d            n# 1 swxY w Y   t          ||           sJ d S )NzCarl Carl Carl Carl Joe Carl)dtype)   r3      r1   r6   r3   i  r6   r1      r   rU   r
      r3      r2      )BuyerQuantityDater\   z:DataFrameGroupBy.resample operated on the grouping columnsFrC   rZ   5D)r   splitobjectr   	set_indexr   r    r   r   resamplesumr   )r   dfr*   rN   s       r.   (test_groupby_resample_preserves_subclassrd   j   s    
:@@BB&QQQ+++q!R++q!R++r1b!,,r1b!,,r1b!,,q!R++	
 	

 
B 
f		B GC		# %	
 
 
 : : G$$--d337799: : : : : : : : : : : : : : : fc"""""""s   ;DD#&D#)r   numpyr   r   pandasr   r   r   pandas._testing_testingr   pandas.tests.groupbyr   markfilterwarnings
pytestmarkparametrizer=   r   r#   r/   rQ   rd    rB   r.   <module>ro      sn                      
       8 8 8 8 8 8[''S 

 	YRYq"%5%5677IBIa,,3777 1 1 1D+- +- +-\ B,B CDD# # ED# # #rB   