
    dw                         d dl Z d dlmc mZ d dlmZmZ d dlm	Z
  ej        d          e j                            d           G d d                                  ZdS )    N)	DataFrameSeriesnumbaignorec                   ,    e Zd Zd Zd Zd Zd Zd ZdS )
TestEnginec                 >   |\  }}t          g dt          d          t          dd          d          }|||d}	|                    d|          }
 t          |
|          dd	|	d
|} t          |
|          di |}|dv}t	          j        |||           d S )N      r   r            abcnogilparallelnopythonr   sortr   engineengine_kwargssumminmaxcheck_dtype )r   rangegroupbygetattrtmassert_frame_equalselfr   r   r   r   numba_supported_reductionsfunckwargsdfr   gbresultexpectedr"   s                 [/var/www/html/t/fyr/venv311/lib/python3.11/site-packages/pandas/tests/groupby/test_numba.pytest_cython_vs_numba_framez%TestEngine.test_cython_vs_numba_frame   s     2f\\\auQ{{KKLL"'X8TTZZ$Z''"T"" 
-
 
;A
 
 %72t$$..v.."77
fhKHHHHHH    c                 J   |\  }}t          g dt          d          t          dd          d          }|||d}	|                    d|          d	         }
 t          |
|          dd
|	d|} t          |
|          di |}|dv}t	          j        |||           d S )Nr
   r   r   r   r   r   r   r   r   r   r   r   r!   r#   )r   r$   r%   r&   r'   assert_series_equalr)   s                 r2   test_cython_vs_numba_getitemz'TestEngine.test_cython_vs_numba_getitem   s     2f\\\auQ{{KKLL"'X8TTZZ$Z'',"T"" 
-
 
;A
 
 %72t$$..v.."77
vx[IIIIIIr4   c                     |\  }}t          t          d          g dd          }|||d}	|                    d|          }
 t          |
|          dd|	d	|} t          |
|          di |}|d
v}t	          j        |||           d S )Nr   )r   r   r   foo)indexnamer   r   )levelr   r   r   r   r!   r#   )r   r$   r%   r&   r'   r6   )r*   r   r   r   r   r+   r,   r-   serr   r/   r0   r1   r"   s                 r2   test_cython_vs_numba_seriesz&TestEngine.test_cython_vs_numba_series.   s     2fU1XXYYYU;;;"'X8TT[[qt[,,"T"" 
-
 
;A
 
 %72t$$..v.."77
vx[IIIIIIr4   c                 6   |\  }}t          g dt          d          t          dd          d          }|                    dd          }t          j        t
          d	
          5   t          ||          dddi| d d d            d S # 1 swxY w Y   d S )Nr
   r   r   r   r   r   F)as_indexzas_index=Falsematchr   r   r#   r   r$   r%   pytestraisesNotImplementedErrorr&   r*   r+   r,   r-   r.   r/   s         r2   test_as_index_false_unsupportedz*TestEngine.test_as_index_false_unsupported=   s    1f\\\auQ{{KKLLZZeZ,,].6FGGG 	8 	8GB77W7777	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8   (BBBc                 6   |\  }}t          g dt          d          t          dd          d          }|                    dd          }t          j        t
          d	          5   t          ||          dd
di| d d d            d S # 1 swxY w Y   d S )Nr
   r   r   r   r   r   )axiszaxis=1rA   r   r   r#   rC   rG   s         r2   test_axis_1_unsupportedz"TestEngine.test_axis_1_unsupportedD   s    1f\\\auQ{{KKLLZZ!Z$$].h??? 	8 	8GB77W7777	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8rI   N)__name__
__module____qualname__r3   r7   r>   rH   rL   r#   r4   r2   r   r      sh        I I IJ J JJ J J8 8 88 8 8 8 8r4   r   )rD   pandas.util._test_decoratorsutil_test_decoratorstdpandasr   r   pandas._testing_testingr'   
skip_if_nomarkfilterwarningsr   r#   r4   r2   <module>rZ      s     ) ) ) ) ) ) ) ) )              wH%%:8 :8 :8 :8 :8 :8 :8 &% :8 :8 :8r4   