gL i^ddlZddlZddlmZddlmcmZddl Z ddl m Z m Z ddl mZddlmZej$dej&j(ej&j+gZej.dZej3ej&j+eej4edk(xredej4ej6dd g d Zd Zd ZdZej&jAdddddgdZ!ej&jAdgdgdgdZ"dZ#dZ$dZ%y)N)is_platform_arm) DataFrameIndex)Versionnumbaz0.61z&Segfaults on ARM platforms with numba )reason)paramsc|jSN)param)requests c/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/pandas/tests/apply/test_numba.py apply_axisrs ==cd}|j|d|}|j|d|}tj||y)Nc|Sr xs rz+test_numba_vs_python_noop.. Qrrengineaxispython)applytmassert_frame_equal) float_framerfuncresultexpecteds rtest_numba_vs_python_noopr$sE D   tG*  EF  hZ HH&(+rc dtdtddgtjtj tddgtjtj }d }|j |d d }|j |d d }tj||ddy)Nr ab)na_valuedtypery)indexcolumnsc|Sr rrs rrz3test_numba_vs_python_string_index..-rrrrrrF)check_column_typecheck_index_type) rrpd StringDtypenpnanrrr)dfr!r"r#s r!test_numba_vs_python_string_indexr6&s  S#Jbnnbff&EFsCj(GH B D XXd7X 3FxxXAx6HEErcLtgdgdgddtgd}d}|j|dd }|j|d d }tj||d }|j|dd }|j|d d }tj||y)Nr )g@g @g"@r&r'c)ABCr,c |dS)Nr@rrs rrz/test_numba_vs_python_indexing..: 3rrr rrc |dS)NrArrs rrz/test_numba_vs_python_indexing..?rFrr)rrrrassert_series_equal)framerow_funcr"r#col_funcs rtest_numba_vs_python_indexingrL5s io>O$ E H [['[ :F{{8H1{=H68,H [['[ :F{{8H1{=H68,r reductionc"|jSr )meanrs rrrGs qvvxrc"|jSr )minrs rrrGs 1557rc"|jSr )maxrs rrrGs aeegrc"|jSr )sumrs rrrGs rcttjdtj}|j |d|}|j |d|}t j ||y)N)r<r<r)rrr)rr3onesfloat64rrrH)rMrr5r"r#s rtest_numba_vs_python_reductionsrYEsU 27764 5B XXijX AFxx (xDH68,rcolnamesr8)g?g@g@cttjgdgdgdgtj|}|dfd}|j |dd }|j |d d }t j ||y) Nr8r;) r))r-rc|Sr r)r first_cols rrz-test_numba_numeric_colnames..Ws !I,rrr rr)rr3arrayint64rrrH)rZr5fr"r#r`s @rtest_numba_numeric_colnamesrdPsr  )Y 2"((CX B IA XXaaX 0Fxx(x3H68,rcd}tjtd5|j|dddidddy#1swYyxYw)Nc|Sr rrs rrz1test_numba_parallel_unsupported..^!rzAParallel apply is not supported when raw=False and engine='numba'matchrparallelT)r engine_kwargs)pytestraisesNotImplementedErrorr)r rcs rtest_numba_parallel_unsupportedro]sNA Q O !GJ;MN OOOs ?Acd}tdddgitddg}tjtd5|j |d| dddy#1swYyxYw) Nc|Sr rrs rrz2test_numba_nonunique_unsupported..grgrr&r r9rDzBThe index/columns must be unique when raw=False and engine='numba'rhrr)rrrlrmrnrrrcr5s r test_numba_nonunique_unsupportedrsfs_A C!Q=sCj(9 :B R 5 74 555s AA"ctjdd}tddgddgddgd }|d jd |d <tjt d 5|j |d|dddtjt d 5|d jj |d|dddy#1swYSxYw#1swYyxYw)Npyarrowc|Sr rrs rrz/test_numba_unsupported_dtypes..rrgrr r9r&r'r<r=r?r@zdouble[pyarrow]z>Column b must have a numeric dtype. Found 'object|str' insteadrhrrzUColumn c is backed by an extension array, which is not supported by the numba engine.)rl importorskiprastyperm ValueErrorrto_framerrs rtest_numba_unsupported_dtypesr{ps  "A !QsCj1v> ?Bgnn./BsG N 5 74 5 6 E 3  7 D EE 55 EEs C&CCC)&numpyr3rl pandas.compatrpandas.util._test_decoratorsutil_test_decoratorstdpandasr1rrpandas._testing_testingrpandas.util.versionr skip_if_nomark single_cpuskipif pytestmarkrwrappend __version__fixturerr$r6rL parametrizerYrdrorsr{rrrrs\ )))'bmmG$fkk&<&P>P>R S G$  KK!!"gfo5K/:K78I8I7JK1v,  - *,=?PQ- -i%AB -C -O5Er