
    I jr                        S SK rS SKrS SKrS SKJr  \R                  R                  SS\R                  " SS94S/\R                  " SS9/4SS/S\R                  " SS9/4SS/\R                  " SS9S/4/5      \R                  R                  SS/SS//5      S 5       5       r\R                  R                  SSSSS/S/S/SS/SS// S	Q/ S
Q/
5      S 5       rg)    Nzkey_strs,groupersinnerlevelBlevelsouterc                    [         R                  " / SQ/ SQ[        R                  " S5      / SQS.5      nUR	                  U 5      nSU;  d  SUR
                  ;   a=  UR                  U5      R                  SS	9nUR                  U5      R                  SS	9nO>UR                  U5      R                  5       nUR                  U5      R                  5       n[        R                  " XE5        g )
Nar   r   br   r            r   r   r      oner   twor   r   r   r   r   Ar   r   r   T)numeric_only)
pd	DataFramenparange	set_indexcolumnsgroupbymeantmassert_frame_equal)r   key_strsgroupersframeresultexpecteds         z/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/pandas/tests/groupby/test_index_as_string.py"test_grouper_index_level_as_stringr(      s     LL3'1;		
E OOF#E
(g6x(--4-@==*//T/Bx(--/==*//1&+    )r   r   r   )r   r   r   c                    [         R                  " / SQ/ SQ[        R                  " S5      / SQS.5      nUR	                  / SQ5      S   n[        U [        5      (       a%  U  Vs/ s H  n[         R                  " US9PM     nnO[         R                  " U S9nUR                  U5      R                  5       nUR                  U 5      R                  5       n[        R                  " Xe5        g s  snf )	Nr
   r   r   r   r   )r   r   r   r   r   )r   r   r   r   r   
isinstancelistGrouperr   r   r    assert_series_equal)r   dfserieslvr#   r&   r%   s          r'   )test_grouper_index_level_as_string_seriesr2   %   s    " 
3'1;		

B \\1237F&$39:6RBJJR(6:::F+~~h',,.H ^^F#((*F6, ;s   %C/)numpyr   pytestpandasr   pandas._testing_testingr    markparametrizer-   r(   r2    r)   r'   <module>r;      s       	"**7+,
RZZg./0
w#rzz89:
3"**73S9:	 WI/A#BC, D,& 						'	'--r)   