Ë
|
nñúh^ ã ó¼ d dl Zd dlZd dlmZ d dlmc mZ d dl Z
|
d dl m Z m Z d dl mZ d dlmZ ej$ d« ej& j( ej& j+ « gZ ej. d« Zej3 ej& j+ eej4 « ed« k( xr e« dej4 ¬« « ej6 d d g¬
|
« d « Z d Z d Z d Z ej& jA dd d d d g« d « Z!ej& jA dg d¢g d¢g« d « Z"d Z#d Z$d Z%y) é N)Úis_platform_arm)Ú DataFrameÚIndex)ÚVersionÚnumbaz0.61z&Segfaults on ARM platforms with numba )Úreasoné )Úparamsc ó | j S ©N)Úparam)Úrequests úPH:\Change_password\venv_build\Lib\site-packages\pandas/tests/apply/test_numba.pyÚ
|
apply_axisr s à ==Ðó c ó d }| j |d|¬« }| j |d|¬« }t j ||« y )Nc ó | S r © ©Úxs r ú<lambda>z+test_numba_vs_python_noop.<locals>.<lambda> ó Q r r ©ÚengineÚaxisÚpython)ÚapplyÚtmÚassert_frame_equal)Ú float_framer ÚfuncÚresultÚexpecteds r Útest_numba_vs_python_noopr$ sE Ù DØ × Ñ t¨G¸*Ð Ó EFØ× Ñ ¨h¸ZÐ ÓHHÜ×Ñ& (Õ+r c
|
ód t dt ddgt j t j
|
¬« ¬« t ddgt j t j
|
¬« ¬« ¬« } d }| j |d
|
d ¬ « }| j |d d ¬ « }t j ||dd¬« y )Nr ÚaÚb)Úna_value©Údtyper Úy)ÚindexÚcolumnsc ó | S r r r s r r z3test_numba_vs_python_string_index.<locals>.<lambda>- r r r r r r F)Úcheck_column_typeÚcheck_index_type) r r ÚpdÚ StringDtypeÚnpÚnanr r r )Údfr! r" r# s r Ú!test_numba_vs_python_string_indexr6 & s ä Ø ÜS#J¤b§n¡n¼b¿f¹fÔ&EÔFÜsCj¬¯©ÄÇÁÔ(GÔHô
|
Bñ
|
DØ XXd 7°XÓ 3FØxx X°AxÓ6HÜ×ÑبEÀEör c óL t g d¢g d¢g d¢dt g d¢« ¬« } d }| j |dd ¬
|
« }| j |d d ¬
|
« }t j ||« d }| j |dd ¬
|
« }| j |d d ¬
|
« }t j ||« y )N©r é é ©é é é )g @g @g "@©r& r' Úc)ÚAÚBÚC©r, c ó | d S )Nr@ r r s r r z/test_numba_vs_python_indexing.<locals>.<lambda>: ó
|
3 r r r r r c ó | d S )NrA r r s r r z/test_numba_vs_python_indexing.<locals>.<lambda>? rF r r )r r r r Úassert_series_equal)ÚframeÚrow_funcr" r# Úcol_funcs r Ú test_numba_vs_python_indexingrL 5 s Ü ÚiªoÑ>ÜOÓ$ô Eñ HØ [[¨'¸[Ó :FØ{{8¨H¸1{Ó=HÜ×Ñ6 8Ô,á HØ [[¨'¸[Ó :FØ{{8¨H¸1{Ó=HÜ×Ñ6 8Õ,r Ú reductionc ó" | j « S r )Úmeanr s r r r G s qvvx r c ó" | j « S r )Úminr s r r r G s 1§5¡5£7 r c ó" | j « S r )Úmaxr s r r r G s °a·e±e³g r c ó" | j « S r )Úsumr s r r r G s ÈÏÉË r c óÜ t t j dt j ¬« « }|j | d|¬« }|j | d|¬« }t j ||« y )N)r< r< r) r r r )r r3 ÚonesÚfloat64r r rH )rM r r5 r" r# s r Ú test_numba_vs_python_reductionsrY E sU ô
|
|
2776¬¯©Ô4Ó 5BØ XXi¨°jXÓ AFØxx ¨(¸xÓDHÜ×Ñ6 8Õ,r Úcolnamesr8 )g ð?g @g @c ó t t j g d¢g d¢g d¢gt j ¬« | ¬« }| d fd}|j |dd ¬
|
« }|j |d d ¬
|
« }t j ||« y ) Nr8 r; )é é é r) )r- r c ó | S r r )r Ú first_cols r r z-test_numba_numeric_colnames.<locals>.<lambda>W s ø !I, r r r r r )r r3 ÚarrayÚint64r r rH )rZ r5 Úfr" r# r` s @r Útest_numba_numeric_colnamesrd P sr ø ô
|
Ü
|
)Yª Ð2¼"¿(¹(ÔCÈXô
|
Bð IÛ AØ XXa ¨aXÓ 0FØxx (°xÓ3HÜ×Ñ6 8Õ,r c ó d }t j t d¬« 5 | j |dddi¬« d d d « y # 1 sw Y y xY w)Nc ó | S r r r s r r z1test_numba_parallel_unsupported.<locals>.<lambda>^ ó ! r zAParallel apply is not supported when raw=False and engine='numba'©Úmatchr ÚparallelT)r Ú engine_kwargs)ÚpytestÚraisesÚNotImplementedErrorr )r rc s r Ú test_numba_parallel_unsupportedro ] sN ÙAÜ ÜØQô
|
ñ Oð ×Ñ! G¸JÈÐ;MÐÔN÷ O÷ Oñ Oús ?¿Ac óÊ d }t dddgit ddg« ¬« }t j t d¬« 5 |j |d| ¬ « d d d « y # 1 sw Y y xY w)
|
Nc ó | S r r r s r r z2test_numba_nonunique_unsupported.<locals>.<lambda>g rg r r& r r9 rD zBThe index/columns must be unique when raw=False and engine='numba'rh r r )r r rl rm rn r ©r rc r5 s r Ú test_numba_nonunique_unsupportedrs f s_ ÙAÜ C!Q=¬¨s°C¨jÓ(9Ô :BÜ ÜØRô
|
ñ 5ð 7¨Ô4÷ 5÷ 5ñ 5ús »AÁA"c ó¾ t j d« d }t ddgddgddgd « }|d
|
j d « |d
|
< t j t
|
d ¬ « 5 |j |d| ¬« d d d « t j t
|
d¬ « 5 |d
|
j « j |d| ¬« d d d « y # 1 sw Y SxY w# 1 sw Y y xY w)NÚpyarrowc ó | S r r r s r r z/test_numba_unsupported_dtypes.<locals>.<lambda>r rg r r r9 r& r' r< r= r? r@ zdouble[pyarrow]z>Column b must have a numeric dtype. Found 'object|str' insteadrh r r zUColumn c is backed by an extension array, which is not supported by the numba engine.)rl Ú importorskipr Úastyperm Ú
|
ValueErrorr Úto_framerr s r Ú test_numba_unsupported_dtypesr{ p sÝ Ü
|
×Ñ Ô"ÙAÜ !Q s¨C j¸¸1°vÑ>Ó ?BØgnnÐ .Ó/BsGä ÜØNô
|
ñ 5ð 7¨Ô4÷ 5ô
|
Üð6ô
|
ñ Eð
|
3×ÑÓ× Ñ ¨7¸Ð ÔD÷ Eð E÷ 5ð 5ú÷ Eð Eús Á CÂ&CÃCÃC )&Únumpyr3 rl Ú pandas.compatr Ú pandas.util._test_decoratorsÚutilÚ_test_decoratorsÚtdÚpandasr1 r r Úpandas._testingÚ_testingr Úpandas.util.versionr Ú
|
skip_if_noÚmarkÚ
|
single_cpuÚskipifÚ
|
pytestmarkrw r ÚappendÚ __version__Úfixturer r$ r6 rL Ú parametrizerY rd ro rs r{ r r r ú<module>r s\ ðÛ Û å )ß )Ð )ã ÷õ Ý 'àbmmGÓ$ f§k¡k×&<Ñ&<¸f¿k¹k×>PÑ>PÓ>RÐ S
|
à×ÑGÓ$Ø
|
× Ñ Ø
|
KK×ÑÙ×!Ñ!Ó"¡g¨f£oÑ5ÒK¹/Ó:KØ7¸×8IÑ8IÐ7JÐKð ó ôð 1vÔ ñó ðò,ò ò -ð ×ÑØÙÑ*Ñ,=Ñ?PÐQóñ-ó ð-ð ×Ñ¢i² Ð%AÓBñ -ó Cð -òOò5óEr
|
