
    d              
       r   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 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_argsobjA
   namec                    t          | t                    r|dv rt          j        d|            |                     t          j        dd                    }t          |                    d          t          |                     sJ t          ||           } t          ||          | } |j        |g|R  }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
   >   sizengroupcumcount)
isinstancer   pytestskipgroupbynparange	get_grouptyper   getattraggr   tmSubclassedSeriesassert_frame_equalassert_series_equal)r   groupby_funcgroupedargsresult1result2slicess          f/var/www/html/t/fyr/venv311/lib/python3.11/site-packages/pandas/tests/groupby/test_groupby_subclass.pytest_groupby_preserves_subclassr'      sh    #v E<<#?#?C\CCDDDkk")Ar**++G g''**DII66666"<55D,gg|,,d3Ggk,....G ,++F#y!! .lf&<&<'2#6777777'499----- '9%% 1
gw/////
w00000    c                     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        dgd	z  t          g dd
                    }t          j        ||           d }t          j        g d          }d|_        |                    | d                                       |          }t          j        ||           |                    | d                   	                    |          }t          j        ||           d S )N)         )r*   r*   r+   )      	   )abctestattrhellor2   c                 l    t          | t          j                  sJ t          | d          sJ | j        S Nr3   )r   r   SubclassedDataFramehasattrr3   groups    r&   funcz-test_groupby_preserves_metadata.<locals>.func=   s9    %!788888uj)))))~r(   r,   r   )indexc                 l    t          | t          j                  sJ t          | d          sJ | j        S r6   )r   r   r   r8   r3   r9   s    r&   func2z.test_groupby_preserves_metadata.<locals>.func2F   s9    %!455555uj)))))~r(   )
r   r7   	_metadatar3   r   applyr   r   r   r   )	custom_df_group_dfr;   resultexpectedr>   custom_seriess           r&   test_groupby_preserves_metadatarG   4   s   &YYYYYYYYY'W'WXXI,,,,, I ((-- , ,8 G+++++  
 s##))$//F"G9q=iiic8R8R8RSSSH68,,,  
 '			22M$M""9S>2288??F68,,,""9S>2266u==F68,,,,,r(   c                     | d                                 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                              d                                          }t          ||           sJ d S )NzCarl Carl Carl Carl Joe Carl)   r,      r*   r/   r,   i  r/   r*      r   rJ   r
      r,      r+      )BuyerQuantityDaterQ   rO   5D)splitr   	set_indexr   resamplesumr   )r   dfrD   s      r&   (test_groupby_resample_preserves_subclassrX   S   s
    
399;;+++q!R++q!R++r1b!,,r1b!,,r1b!,,q!R++	
 	

 
B 
f		B ZZ  ))$//3355Ffc"""""""r(   )r   numpyr   r   pandasr   r   r   pandas._testing_testingr   pandas.tests.groupbyr   markparametrizer7   r   r   r'   rG   rX    r(   r&   <module>ra      sR                      
       8 8 8 8 8 8 	YRYq"%5%5677IBIa,,3777 1 1 1<- - -> B,B CDD# # ED# # #r(   