U
|
ý°djo ã @ s, d Z ddlmZmZmZmZ ddlmZ ddlmZm Z ddl
|
Z
|
ddl Z ddl Z ddl ZddlZddlmZ ddlmZmZmZmZmZmZmZ ddlmZ dd l m Z dd
|
l m Z ddl Z!ddl"m#Z$ dd l%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z, dd l-m.Z. ej/d ddZ0ddl1m2Z2 ddl3m4Z4 ej5e$j6e7ddd Z8ej5dd Z9ej5dd Z:ej5ddgddd Z;ej5d d Z<ej5d d Z=ej5d d! Z>ej5d"d# Z?ej5d$d% Z@G d&d' d'e.jAZBG d(d) d)e.jCZDG d*d+ d+e.jEZFG d,d- d-e.jGZHG d.d/ d/e.jIZJG d0d1 d1e.jKZLG d2d3 d3e.jMZNG d4d5 d5e.jOZPG d6d7 d7e.jQZRG d8d9 d9e.jSZTG d:d; d;e.jUZVG d<d= d=e.jWZXG d>d? d?e.jYZZG d@dA dAe.j[Z\G dBdC dCe.j]Z^G dDdE dEe.j_Z`G dFdG dGe.jaZbG dHdI dIe.jcZdG dJdK dKe.jeZfG dLdM dMZgdNdO ZhdPdQ Ziejj kdRdSdTdUdVdWg¡ejj kdXdYdYdYgg¡dZd[ Zlejjjkd\ddd]d]d^d^gdd^ggdddd]d^d^gdgggd_d`gdadbdc Zmddde Znejj kdfe0 o¡ epge0 odg¡epge0 q¡ epge0 r¡ e7ge0 se0 t¡ ¡euge0 ve0 t¡ ¡euge0 we0 x¡ e0 t¡ ¡eyge0 ze0 t¡ e0 t¡ ¡e gg¡dhdi Z{djdk Z|dldm Z}dndo Z~dpdq Zdrds Zdtdu Zdvdw Zdxdy Zdzd{ Zd|d} Z
d~d Zdd Zejj kdejejfdg¡dd
Zdd Zdd Zdd Zejjjedddd Zejjjedddd Zdd Zdd Zejj kdddg¡dd Zdd Zejj kddd gd¡d¢gd£d¤gg¡d¥d¦ Zd§d¨ Zejj kd©dªd«dd«d¬dggdd¬dd«d«dggdªd«d¬d«d¬d¬ggd®d«dd¬d¬dggd¯d¬dd¬d«dggg¡d°d± Zd²d³ Zejj kd´dµdªdd¬dggdµd¶d«d«d«ggd·d¶d¬d«d¬ggd·d¸dd¬dggg¡d¹dº Zejj kd»de d¶¡gd¼e7gd½d«gd¾d¿gg¡dÀdÁ Zejj kdÂdÃdÄdÅd«dÆdggdÃdÄd¿d«dÇdggdÈdÄdÅd¬dÉdggg¡dÊdË ZdÌdÍ ZejjjedÎedÏdÐdÑ Z ejj kdÒdªd«dd¬dggdd¬dd«dggd¸d¬dd«dggdªd«d¬d¬d¬ggd®d«dd¬dggd¯d¬dd«dggg¡dÓdÔ Z¡ejj kdÒdd«dd¬dggdÕd¬dd«dggd¸d¬dd«dggdªd«d¬d¬d¬ggdd«dd¬dggd¯d¬dd«dggg¡dÖd× Z¢ejj kdØdªddddge0 £¡ gd¸d¿dÙd]dge0 t¡ gg¡dÚdÛ Z¤dÜdÝ Z¥ejj kdÞd¿d¶dßdggdÅdàdßdggd]dàddggdádÃddggd^dddggg¡dâdã Z¦ejjjdäe§dÏdådæ Z¨ejj kdçdd]ddªdggdd]d¿dªdggd¿dÙd¿d¸dggg¡dèdé Z©ejj kdêd¿d]dÄdëdggdd]dÄdìdggdd]ddídggg¡dîdï Zªejj kdðdñdòd¬gdódòd«gdôdõd¬gdödõd«gd÷død¬gdùdød«gdúdûd¬gdüdûd«gdôdýd¬gdþdýd«gdÿd d¬gdd d«gddd¬gddd«gddd¬gddd«gdd d¬gd
|
d d«gg¡d d Z«ejj kd ddgddgddgdTdgddgddgg¡dd Z¬dd Zejj kd d dd gd dÄd gd dd!gd dÄd"gd#dd$gd#dÄd%gg¡d&d' Z®ejj kd(d)dg¡d*d+ Z¯ejj kd(d,dg¡d-d. Z°ejj kd/d0d1g¡ejj kd2d3d4gd5d6gg¡d7d8 Z±ejj kd¾dd¿g¡d9d: Z²ejj kd;d<d=g¡ejj kd>ddgd¿d^gg¡d?d@ Z³ejj kdAdBdCg¡dDdE Z´ejj kd>ddgd¿d^gg¡dFdG ZµdHdI Z¶dJdK Z·dLdM Z¸dNdO Z¹dPdQ ZºdRdS Z»dTdU Z¼ejj kdVdWdXdYdZg¡d[d\ Z½d]d^ Z¾ejj kd_d`dagdbd]gdcdgdddgdedgdfd]gdgd]gdhdÙgdid^gej¿djd«ejjjeedkdldmdndogdpd¿gdqdrgdsd¿gdtdugdvedad¿d]gdwedÙd^dudogg¡dxdy ZÀejj kdVdXdWg¡dzd{ ZÁejj kd|dd}d~g¡dd ZÂdd ZÃdd ZÄejj kd;d
ddg¡dd ZÅejj kd;d
ddg¡dd ZÆejjjeddejj kddddddddg¡ejj kd;d
ddg¡dd ZÇdd ZÈdd ZÉdd ZÊejj kdVdXdWg¡dd ZËejj kdd edadÙd¡dÙd¢gd£e! Ìd¤¡gg¡d¥d¦ ZÍejj kd§d¬d«g¡d¨d© ZÎejj kddªd«g¡d¬d ZÏd®d¯ ZÐejj kd°e$jÑe$jÒ ¡d±d² ZÓejjjed³e0jÔdÏd´dµ ZÕdS (¶ ag
|
This file contains a minimal set of tests for compliance with the extension
|
array interface test suite, and should contain no other tests.
|
The test suite for the full functionality of the array is located in
|
`pandas/tests/arrays/`.
|
The tests in this file are inherited from the BaseExtensionTests, and only
|
minimal tweaks should be applied to get the tests passing (by overwriting a
|
parent method).
|
Additional tests should either be added to one of the BaseExtensionTests
|
classes (if they are relevant for the extension interface for all dtypes), or
|
be added to the array-specific tests in `pandas/tests/arrays/`.
|
é )ÚdateÚdatetimeÚtimeÚ timedelta)ÚDecimal)ÚBytesIOÚStringION)Úlib)ÚPY311Úis_ci_environmentÚis_platform_windowsÚpa_version_under7p0Úpa_version_under8p0Úpa_version_under9p0Úpa_version_under11p0)ÚPerformanceWarning)Úis_any_int_dtype)ÚCategoricalDtypeType)Ú is_bool_dtypeÚis_float_dtypeÚis_integer_dtypeÚis_numeric_dtypeÚis_signed_integer_dtypeÚis_string_dtypeÚis_unsigned_integer_dtype)ÚbaseZpyarrowz7.0.0)Z
|
minversion)ÚArrowExtensionArray)Ú
|
ArrowDtype)ÚparamsÚidsc C s t | jdS )N)Ú pyarrow_dtype)r Úparam)Úrequest© r# úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/test_arrow.pyÚdtypeB s r% c C s| | j }tj |¡rBddgd d g ddgd d g ddg }n,tj |¡r~ddgd d g ddgd d g d d
|
g }nðtj |¡rºd d gd d g d dgd d g d dg }n´tj |¡röd d gd d g dd gd d g d dg }nxtj |¡rLtdtdgd d g tdtdgd d g tdtdg }n"tj |¡rºt
|
dd d t
|
dddgd d g t
|
dd d t
|
dd d gd d g t
|
dddt
|
dddg }n´tj |¡r<t dd d d d d d t dd d d d d d gd d g t dd d d t dd d d gd d g t dd d t dd d g }n2tj |¡rtd td d gd d g tdtd gd d g td td g }nÜtj |¡ròtdd td dgd d g td d td d gd d g td d td d g }n|tj |¡r.d d gd d g dd!gd d g d"d#g }n@tj |¡rjd$d%gd d g d&d'gd d g d(d)g }nttj|| d*S )+NTFé é, ç ð?ç g Àg ð¿ç à?g àX@é r éþÿÿÿéÿÿÿÿéc é Ú1ú0.0z-2.0z-1.0z0.5z33.123éæ éÏ é é éä iöÿÿÿé
|
é ÚaÚbÚ2ú!ú>ó aó bó 1ó 2ó !ó >©r% )r ÚpaÚtypesÚ
|
is_booleanÚ is_floatingÚis_signed_integerÚis_unsigned_integerÚ
|
is_decimalr Úis_dater Ú is_timestampr Ú is_durationr Úis_timer Ú is_stringÚ is_binaryÚNotImplementedErrorÚpdÚarray)r% Úpa_dtypeÚdatar# r# r$ rV G s 0 0 0 0ÿþýüÿÿþýüÿ*ÿ þýüÿÿþýüÿÿþýüÿ..rV c C s t | d| d g¡S )z Length-2 array with [NA, Valid]Nr ©ÚtypeÚ_from_sequence)rV r# r# r$ Ú data_missing s rZ )r c C s | j dkr|S | j dkr |S dS )zParametrized fixture returning 'data' or 'data_missing' integer arrays.
|
|
Used to test dtype conversion with and without missing values.
|
rV rZ N)r! )r" rV rZ r# r# r$ Úall_data s
|
|
r[ c C s¾ | j }tj |¡r"d}d}d}n~tj |¡r>d}d}d}nbtj |¡rZd}d}d}nFtj |¡rvd}d}d }n*tj |¡r¨td
|
d d }td dd}tddd}nøtj |¡rät
|
d
|
dddddd}t
|
ddd}t
|
dddd}n¼tj |¡rt d}t d}t dd}ntj |¡r<tdd}tdd }td d }ndtj |¡rXd}d}d}nHtj |¡rtd}d}d}n,tj |¡rtd}td}td}nttj||dd||||g| dS ) z
|
Data for factorization, grouping, and unique tests.
|
|
Expected to be like [B, B, NA, NA, A, A, B, C]
|
|
Where A < B < C and NA is missing
|
FTgñ¿r) gñ?r- r r+ r7 r3 r4 r5 iÚ r2 r6 r& r9 r: Úcr> r? ó cz-1.1r1 z1.1NrD )r rE rF rG rH rI rJ rL r rM r rN r rO r rP rQ rK r rR rS rT )r% rU ÚAÚBÚCr# r# r$ Údata_for_grouping s^
|
|
|
ra c C s t | | d | d | d g¡S )zn
|
Length-3 array with a known sort order.
|
|
This should be three items [B, C, A] with
|
A < B < C
|
r é r& rW ©ra r# r# r$ Údata_for_sortingÐ s ÿrd c C s t | | d | d | d g¡S )z{
|
Length-3 array with a known sort order.
|
|
This should be three items [B, NA, A] with
|
A < B and NA missing.
|
r r/ r& rW rc r# r# r$ Údata_missing_for_sortingÝ s ÿre c C s: | j j}tj |¡s tj |¡r6tjdgd | j dS | S )z3Length-100 array in which all the elements are two.r/ éd rD )r% r rE rF Ú
|
is_integerrH rS rT ©rV rU r# r# r$ Ú data_for_twosê s ri c C s t jS )z6The scalar missing value for this type. Default 'None')rS ÚNAr# r# r# r$ Úna_valueô s rk c s e Zd Z fddZ ZS )ÚTestBaseCastingc sB |j j}tj |¡r2|j tjj d| dd¡ t
|
|¡ d S )NzFor z .astype(str) decodes.©Úreason) r% r rE rF rQ ÚnodeÚ
|
add_markerÚpytestÚmarkÚxfailÚsuperÚtest_astype_str©ÚselfrV r" rU ©Ú __class__r# r$ ru û s
|
ÿÿz TestBaseCasting.test_astype_str)Ú__name__Ú
|
__module__Ú __qualname__ru Ú __classcell__r# r# rx r$ rl ú s rl c s4 e Zd Z fddZdd Zdd Zdd Z ZS ) ÚTestConstructorsc sb |j j}tj |¡s tj |¡rRtj |¡r2d}n
|
d| }|j tj j
|
|d¡ t |¡ d S )Nz1ArrowDtype(pa.string()) != StringDtype('pyarrow')ú$pyarrow.type_for_alias cannot infer rm ) r% r rE rF rP rK ro rp rq rr rs rt Útest_from_dtype)rw rV r" rU rn rx r# r$ r s
|
ÿÿz TestConstructors.test_from_dtypec C sd t | |j¡}t ||¡ t|jtjs.tt | |j ¡ ¡}t ||¡ t|jtjs`td S ©N)
|
rX rY Ú_dataÚtmÚ assert_extension_array_equalÚ
|
isinstancerE Z ChunkedArrayÚAssertionErrorÚcombine_chunks©rw rV Úresultr# r# r$ Útest_from_sequence_pa_array s z,TestConstructors.test_from_sequence_pa_arrayc C s2 t jtdd tjdgt ¡ d W 5 Q R X d S )NzConverting strings to©Úmatchz12-1rD )rq ÚraisesrR r Ú_from_sequence_of_stringsrE Zmonth_day_nano_interval)rw r" r# r# r$ Ú*test_from_sequence_pa_array_notimplemented! s
|
ÿz;TestConstructors.test_from_sequence_pa_array_notimplementedc C s |j j}tj |¡r:| d¡r:ts:|j t j
|
j dd¡ nzt rxtj |¡sVtj |¡rx|j t j
|
j tjd| d¡ n<tj |¡r´|jd k r´t r´t r´|j t j
|
j tjdd¡ |j t ¡ ¡}t|j||j d}t ||¡ | ¡ }t|j||j d}t ||¡ d S )Nz
|
time64[ns]z&Nanosecond time parsing not supported.rm z pyarrow doesn't support parsing ©r rn ú_TODO: Set ARROW_TIMEZONE_DATABASE environment variable on CI to path to the tzdata for pyarrow.rD ) r% r rE rF Z is_time64Úequalsr
|
ro rp rq rr rs r rN rK ÚArrowNotImplementedErrorrM Útzr r Ú ArrowInvalidr ÚcastÚstringrX r r r r )rw rV r" rU Zpa_arrayr r# r# r$ Ú&test_from_sequence_of_strings_pa_array' s@ ÿÿ
|
ÿ
|
ÿþÿ ýÿ z7TestConstructors.test_from_sequence_of_strings_pa_array)rz r{ r| r r r r r} r# r# rx r$ r~ s r~ c @ s e Zd ZdS )ÚTestGetitemTestsN©rz r{ r| r# r# r# r$ r L s r c @ sH e Zd Zdd Zej dddg¡dd Zej dddg¡dd Zd
|
S ) ÚTestBaseAccumulateTestsc C sd t |||d}|jjdkr0| d¡}| d¡}| d¡}t | d¡||d}| j||dd d S )N©ÚskipnaÚmúint64[pyarrow]ÚFloat64F)Z check_dtype)Úgetattrr% ÚkindÚastypeÚassert_series_equal)rw ÚserÚop_namer r Úexpectedr# r# r$ Úcheck_accumulateQ s
|
|
|
z(TestBaseAccumulateTests.check_accumulater TFc C sz |j j}tj |¡s,tj |¡s,tj |¡rB|dkrBtsBt d¡ |}t
|
|¡}t t ¡ t|||d W 5 Q R X d S )NÚcumsumz1These work, are tested by test_accumulate_series.r )r% r rE rF rg rH rN r rq ÚskiprS ÚSeriesr rR r¡ )rw rV Úall_numeric_accumulationsr Úpa_typer¦ r¥ r# r# r$ Ú test_accumulate_series_raises] s
|
þ
|
ý
|
üúù
|
|
z5TestBaseAccumulateTests.test_accumulate_series_raisesc C s$ |j j}|}t |¡}d}tj |¡s2tj |¡r@|dkr~d}n>tj |¡rftj |¡sf|dkr~d}ntj |¡r~|dkr~d}|rt
|
d¡ |dkst rÌ|j jjr¬t
|
d¡ |j t
|
jj| dtd ¡ nF|dkrtj |¡sðtj |¡r|j t
|
jj| d
|
| td ¡ | |||¡ d S ) NF)r© ÚcumprodTr¯ z]These should *not* work, we test in test_accumulate_series_raises that these correctly raise.r© zpyarrow xfail slowz not implemented©rn r z not implemented for )r% r rS r« rE rF rP rQ Ú is_temporalrN rq rª r ÚconfigÚoptionZ skip_slowro rp rr rs rR rG rK r¨ ) rw rV r¬ r r" r r¦ r¥ Zdo_skipr# r# r$ Útest_accumulate_seriesq sL
|
ÿ
|
|
þÿ
|
|
ÿ
|
ÿ þÿz.TestBaseAccumulateTests.test_accumulate_seriesN) rz r{ r| r¨ rq rr Ú parametrizer® r´ r# r# r# r$ r P s
|
|
r c sR e Zd Zdd Zej dddg¡ fddZej dd d
|
d g¡d d Z Z S )ÚTestBaseNumericReducec C s |j j}|dkr t|| }nt|||d}tj |¡rFt d¡ n"tj |¡s^tj |¡rh|
|
d¡}|dkr~t|| }nt|||d}t ||¡ d S )NÚcountr zApandas boolean data with NA does not fully support all reductionsr ) r% r r¡ rE rF rG rq rª rg rH r£ r Zassert_almost_equal)rw r¥ r¦ r rU r r§ r# r# r$ Ú check_reduce£ s ÿ
|
z"TestBaseNumericReduce.check_reducer TFc sf |j j}|}t |¡}d}tj |¡rJ|dkrJtj |¡rD|dkrDqnd}n$tj |¡sbtj |¡rn|dkrnd}|s¬d| d} t
|
j t | d t |||d W 5 Q R X d S t
|
jjt | d
|
tj d | d }
|
|d krä|j |
|
¡ nn|dkrtrtj |¡r|j |
|
¡ nB|dkr.tr.|j |
|
¡ n$tj |¡rR|dkrR|j |
|
¡ t |||¡ d S )NT)ÚsumÚvarÚskewÚkurtÚprod)r¹ F) r¹ ÚmeanÚmedianr½ ÚstdÚsemrº r» r¼ z does not support reduction 'ú'r r ú is not implemented in pyarrow=ú for r > r» r¼ > rº r¿ rÀ rÁ > rº r¿ rÀ rÁ )r% r rS r« rE rF r± rN rP rQ rq r Ú TypeErrorr¡ rr rs Ú __version__ro rp r rK r rG rt Útest_reduce_series) rw rV Zall_numeric_reductionsr r" rU Úopnamer¥ Z should_workÚmsgÚ
|
xfail_markrx r# r$ rÇ ¶ sL
|
|
ÿ
|
ÿþ ýÿþ
|
ý z(TestBaseNumericReduce.test_reduce_seriesÚtypÚint64Zuint64Úfloat64c C s, t jddg| dd ¡ }|dks(td S )Nr+ r/ ú [pyarrow]rD ç ø?)rS r« r¿ r )rw rË r r# r# r$ Útest_median_not_approximateý s z1TestBaseNumericReduce.test_median_not_approximate)
|
rz r{ r| r¸ rq rr rµ rÇ rÐ r} r# r# rx r$ r¶ ¢ s
|
Fr¶ c @ s& e Zd Zej dddg¡dd ZdS )ÚTestBaseBooleanReducer TFc C sÆ |j j}tjjt| dtj d| d}tj |¡sBtj
|
|¡rN|j |¡ |}t |¡} tj |¡r¢tj |¡s¢tjtdd t| ||d W 5 Q R X d S t| ||d}
|
|
|
|dkksÂtd S )Nrà rÄ r zdoes not support reductionr r Úany)r% r rq rr rs rÅ rE rÆ rF rP rQ ro rp rS r« r± rN r r¡ r ) rw rV Úall_boolean_reductionsr rk r" rU rÊ r¦ r¥ r r# r# r$ rÇ s ý
|
z(TestBaseBooleanReduce.test_reduce_seriesN)rz r{ r| rq rr rµ rÇ r# r# r# r$ rÑ s rÑ c sR e Zd Z fddZ fddZej dddg¡ fdd Z fd
|
d Z Z
|
S ) ÚTestBaseGroupbyc s@ |j j}tj |¡r0|j tjj | dd¡ t
|
|¡ d S ©Nú" only has 2 unique possible valuesrm ) r% r rE rF rG ro rp rq rr rs rt Ú test_groupby_extension_no_sort©rw ra r" rU rx r# r$ r× ( s ÿÿz.TestBaseGroupby.test_groupby_extension_no_sortc s@ |j j}tj |¡r0|j tjj | dd¡ t
|
|¡ d S rÕ ) r% r rE rF rG ro rp rq rr rs rt Ú test_groupby_extension_transformrØ rx r# r$ rÙ 2 s ÿÿz0TestBaseGroupby.test_groupby_extension_transformÚas_indexTFc sD |j j}tj |¡r2|j tjj t
|
| dd¡ t ||¡ d S )NrÖ r ) r% r rE rF rG ro rp rq rr rs Ú
|
ValueErrorrt Útest_groupby_extension_agg)rw rÚ ra r" rU rx r# r$ rÜ < s þÿz*TestBaseGroupby.test_groupby_extension_aggc s¤ t |jrt ddddddddg|ddddddddgd¡}t dg¡}tjtdd | d ¡ ¡ j
|
W 5 Q R X | d ¡j d
|
d j
|
}t ||¡ n t |¡ d S ) Nr+ r/ é r& )r^ r_ r` r` zdoes not supportr r^ T)Z numeric_only)r r% rS Ú DataFrameZIndexrq r rÅ Úgroupbyr¹ Úcolumnsr Zassert_index_equalrt Útest_in_numeric_groupby)rw ra Údfr§ r rx r# r$ rá H s
|
ýÿ z'TestBaseGroupby.test_in_numeric_groupby) rz r{ r| r× rÙ rq rr rµ rÜ rá r} r# r# rx r$ rÔ ' s
|
|
|
rÔ c sT e Zd Z fddZ fddZ fddZdd Z fd d
|
Z fd d Z Z S ) Ú TestBaseDtypec s@ |j j}tj |¡r0tr0|j tj j
|
t dd¡ t |¡ d S )Nz,decimal string repr affects numpy comparisonr )r% r rE rF rK r ro rp rq rr rs rÛ rt Útest_check_dtyperv rx r# r$ rä \ s þÿz TestBaseDtype.test_check_dtypec sz |j }tj |¡r0|j tjjt d| d¡ tj
|
|¡rjd}tj t |d | |j¡ W 5 Q R X d S t |¡ d S )Nr r z6string\[pyarrow\] should be constructed by StringDtyper )r rE rF rK ro rp rq rr rs rR rP r rÅ Úconstruct_from_stringÚnamert Ú#test_construct_from_string_own_name)rw r% r" rU rÉ rx r# r$ rç g s þÿ z1TestBaseDtype.test_construct_from_string_own_namec sb |j }tj |¡r(t| |j¡r^tn6tj |¡rR|j
|
t j j td| d¡ t |¡ d S )Nr r )r rE rF rP rX Zis_dtyperæ r rK ro rp rq rr rs rR rt Útest_is_dtype_from_name©rw r% r" rU rx r# r$ rè { s þÿz%TestBaseDtype.test_is_dtype_from_namec C s0 d}t jt|d t| d¡ W 5 Q R X d S )Nz*'another_type' must end with '\[pyarrow\]'r Z another_type)rq r rÅ rX rå )rw r% rÉ r# r# r$ Ú.test_construct_from_string_another_type_raises s z<TestBaseDtype.test_construct_from_string_another_type_raisesc s |j }tj |¡sltj |¡sltj |¡r>|jdksl|jd k sltj |¡rT|jdksltj |¡sltj
|
|¡r|j t jj| dd¡ t |¡ d S )NÚnszB does not have associated numpy dtype findable by find_common_typerm )r rE rF rL rO rM Úunitr rN rQ rK ro rp rq rr rs rt Útest_get_common_dtyperé rx r# r$ rí s2
|
ÿ
|
þ
|
üûû
|
ùù
|
ø
|
÷ þÿz#TestBaseDtype.test_get_common_dtypec s0 |j }tj |¡r t|s,tn t |¡ d S r )r rE rF rP r r rt Útest_is_not_string_type)rw r% rU rx r# r$ rî ¦ s z%TestBaseDtype.test_is_not_string_type)
|
rz r{ r| rä rç rè rê rí rî r} r# r# rx r$ rã [ s rã c @ s e Zd ZdS )Ú TestBaseIndexNr r# r# r# r$ rï ® s rï c s, e Zd Zejjddd fddZ ZS )ÚTestBaseInterfacez6GH 45419: pyarrow.ChunkedArray does not support views.F©rn Úrunc s t |¡ d S r )rt Ú test_view©rw rV rx r# r$ ró ³ s zTestBaseInterface.test_view)rz r{ r| rq rr rs ró r} r# r# rx r$ rð ² s
|
ÿrð c s$ e Zd Zdd Z fddZ ZS )ÚTestBaseMissingc C sv || ¡ }|d }| |¡}||k s,t| ||¡ t t¡ |jdd}W 5 Q R X ||k sft| ||¡ d S )Nr Zbackfill)Úmethod)ÚisnaZfillnar r r Zassert_produces_warningr )rw rV Zvalidr r# r# r$ Ú test_fillna_no_op_returns_copy» s
|
z.TestBaseMissing.test_fillna_no_op_returns_copyc s2 t jt|d k dd t ||¡ W 5 Q R X d S )NF)Zcheck_stacklevel)r Zmaybe_produces_warningr rt Útest_fillna_series_method)rw rZ Z fillna_methodrx r# r$ rù Ç s ÿz)TestBaseMissing.test_fillna_series_method)rz r{ r| rø rù r} r# r# rx r$ rõ º s rõ c @ s e Zd ZdS )ÚTestBasePrintingNr r# r# r# r$ rú Î s rú c s, e Zd Zejjddd fddZ ZS )ÚTestBaseReshapingú5GH 45419: pyarrow.ChunkedArray does not support viewsFrñ c s t |¡ d S r )rt Útest_transposerô rx r# r$ rý Ó s z TestBaseReshaping.test_transpose)rz r{ r| rq rr rs rý r} r# r# rx r$ rû Ò s
|
ÿrû c s, e Zd Zejjddd fddZ ZS )ÚTestBaseSetitemrü Frñ c s t |¡ d S r )rt Ú test_setitem_preserves_viewsrô rx r# r$ rÿ Û s z,TestBaseSetitem.test_setitem_preserves_views)rz r{ r| rq rr rs rÿ r} r# r# rx r$ rþ Ú s
|
ÿrþ c @ s& e Zd Zej dddg¡dd ZdS )ÚTestBaseParsingÚenginer\ Úpythonc C s( |j j}tj |¡r.|j tjj t
|
dd¡ ntj |¡r\|j tjj t d| dd¡ nRtj |¡r|jdkr|j tjj tdd¡ n"tj |¡r®|j tjj dd¡ t d tj|t|j d
|
i¡}|jd tjd }tj |¡ròt|}nt|}tj|d t|j i|d }|}| ||¡ d S )NzGH 47534r zParameterized types z not supported.)Úusrë z1https://github.com/pandas-dev/pandas/issues/49767z)CSV parsers don't correctly handle binaryrm Z
|
with_dtyperD F)ÚindexZna_rep)r% r ) r% r rE rF rG ro rp rq rr rs rÅ rK rR rM rì rÛ rQ rS rÞ r« ÚstrZto_csvÚnpÚnanr r Zread_csvÚassert_frame_equal) rw r rV r" rU râ Z
|
csv_outputr r§ r# r# r$ Ú test_EA_typesã sF ÿ
|
þÿþÿ ÿ
|
ÿz TestBaseParsing.test_EA_typesN)rz r{ r| rq rr rµ r r# r# r# r$ r â s r c s e Zd Z fddZ ZS )ÚTestBaseUnaryOpsc sD |j j}tj |¡s4|j tjj tj
|
d| d¡ t |¡ d S )Nz$pyarrow.compute.invert does support r ) r% r rE rF rG ro rp rq rr rs r rt Ú test_invertrv rx r# r$ r
|
s þÿz TestBaseUnaryOps.test_invert)rz r{ r| r r} r# r# rx r$ r
|
s r
|
c s´ e Zd Zej dddg¡ fddZdd Zdd Z fd
|
d Z ej d d ddddde
|
j fdde
|
j fddg¡ fddZ fddZ dZ fdd Z fd d Zd d Z ZS )!ÚTestBaseMethodsÚperiodsr+ r, c sV |j j}tj |¡rD|dkrD|j tjj tj
|
d| d| dd¡ t ||¡ d S )Nr+ z
|
diff with z and periods=z will overflowr ) r% r rE rF rJ ro rp rq rr rs r rt Ú test_diff)rw rV r r" rU rx r# r$ r s ýÿzTestBaseMethods.test_diffc C s. |d d
}| ¡ }|jtt ¡ ks*td S )Nr7 )Ú value_countsr% r rE rÌ r r r# r# r$ Ú'test_value_counts_returns_pyarrow_int64% s z7TestBaseMethods.test_value_counts_returns_pyarrow_int64c C s |d d
¡ }t || ¡ ¡}tj||jd}|jdd ¡ }tjdt | gt | |j
|
dd}| d¡}| ||¡ d S ) Nr7 rD T)Ú normalizer+ Z
|
proportion)r ræ zdouble[pyarrow]) Úuniquer rT r÷ rS r« r% r Z
|
sort_indexÚlenr r£ r¤ )rw rV r" Úvaluesr¥ r r§ r# r# r$ Ú test_value_counts_with_normalize+ s ÿ
|
z0TestBaseMethods.test_value_counts_with_normalizec sv |j j}tj |¡r2|j tjj | dd¡ n0tj
|
|¡rbt rb|j tjj d| tj d¡ t |||¡ d S )NrÖ rm úNo pyarrow kernel for r° )r% r rE rF rG ro rp rq rr rs rK r r rt Útest_argmin_argmax)rw rd re rk r" rU rx r# r$ r 9 s ÿÿþÿz"TestBaseMethods.test_argmin_argmaxzop_name, skipna, expected)ÚidxmaxTr )ÚidxminTr/ )ÚargmaxTr )ÚargminTr/ r Fr )r Fr- )r Fr- c sR |j j}tj |¡r<tr<|r<|j tj j
|
d| tj d¡ t ||||¡ d S )Nr r° )r% r rE rF rK r ro rp rq rr rs r rt Útest_argreduce_series)rw re r¦ r r§ r" rU rx r# r$ r L s þÿ ÿz%TestBaseMethods.test_argreduce_seriesc s@ |j j}tj |¡r0|j tjj | dd¡ t
|
|¡ d S rÕ ) r% r rE rF rG ro rp rq rr rs rt Útest_factorizerØ rx r# r$ r h s ÿÿz TestBaseMethods.test_factorizez bool[pyarrow]c s´ t |djj}tj |¡rPtjjt | d| d}|j
|
|¡ t |¡ n`tj |¡r¤|d\}}t |¡}t |¡}t t ¡ | |dd ¡ W 5 Q R X n t |¡ d S )Nr+ z cannot be added to r r/ c S s | | S r r# )Úx1Zx2r# r# r$ Ú<lambda> ó z2TestBaseMethods.test_combine_add.<locals>.<lambda>)Únextr% r rE rF rN rq rr rs rÅ ro rp rt Útest_combine_addr± rS r« r Úcombine) rw Z data_repeatedr" rU rr Z
|
orig_data1Z
|
orig_data2Ús1Ús2rx r# r$ r" t s þ
|
|
z TestBaseMethods.test_combine_addc sB |j j}tj |¡r0|j tjj | dd¡ t
|
||¡ d S rÕ ) r% r rE rF rG ro rp rq rr rs rt Útest_searchsorted)rw rd Ú as_seriesr" rU rx r# r$ r& s ÿÿz!TestBaseMethods.test_searchsortedc C s t |¡ t |¡¡std S r )rS r« r r rô r# r# r$ Útest_basic_equals s z!TestBaseMethods.test_basic_equals)rz r{ r| rq rr rµ r r r r r r r r Z_combine_le_expected_dtyper" r& r( r} r# r# rx r$ r s.
|
|
øþ
|
|
r c st e Zd ZeZedd Zdd Zdd Zdd Z d d
|
Z
|
d d Z fd dZ fddZ dd Z fddZ ZS )ÚTestBaseArithmeticOpsc K s t |tjr:|jd d
df jj}|jd d
df jj}n|jj}|jj}tj |¡sbtj |¡rv| d¡}| d¡}t
|
j ||f| d S )Nr zfloat[pyarrow]) r
rS rÞ Úilocr% r rE rF rK r£ r Ú assert_equal)ÚclsÚleftÚrightÚkwargsZ left_pa_typeZ right_pa_typer# r# r$ r+ s
|
|
z"TestBaseArithmeticOps.assert_equalc C s4 | d¡}|dkrdd S |dkr*dd S t |¡S )NÚ_Zrtruedivc S s t || ¡S r )r Údivide©ÚxÚyr# r# r$ r ° r z8TestBaseArithmeticOps.get_op_from_name.<locals>.<lambda>Z rfloordivc S s t || ¡S r )r Z floor_divider2 r# r# r$ r ² r )Ústripr Úget_op_from_name)rw r¦ Z short_opnamer# r# r$ r6 ¬ s
|
z&TestBaseArithmeticOps.get_op_from_namec C sì t j | |||¡}d}t|tjrNd}|jd d
df }|jd d
df j}n
|
|}|j}t |j
|
¡}tj |j ¡rª|jj} t |ttfkr| dkrd} | d| d¡}n | |j¡}t |j
|
|}
|
|rÞtj|
|
|j|jd}n
|
t |
|
¡}|S ) NFTr )ÚsÚmsr z duration[ú])r rà )r ÚBaseArithmeticOpsTestsÚ_combiner
rS rÞ r* r% rE rT Z_valuesrF rN rX r rì r r r r rà r« ) rw ÚobjÚotherÚopr§ Z was_frameZ expected_dataZoriginal_dtypeZ pa_expectedrì Z pd_expectedr# r# r$ Ú_patch_combine¶ s0 ÿ
|
z$TestBaseArithmeticOps._patch_combinec C s. t o,|dkrtj |¡p,|dko,tj |¡S )N©Ú__add__Ú__radd__©Ú__sub__Ú__rsub__)r rE rF rN r± )rw rÈ rU r# r# r$ Ú_is_temporal_supportedÜ s
|
ÿ
|
üz,TestBaseArithmeticOps._is_temporal_supportedc C s~ | ||¡}|dkrt}n`|r$d }nV|dkrJtj |¡sDtj |¡rJd }n0tj |¡svtj |¡svtj |¡svtj }nd }|S )N> Ú__rmod__Ú__mod__r@ )
|
rF rR rE rF rP rQ rH rg rK r )rw rÈ rU Úarrow_temporal_supportedÚexcr# r# r$ Ú_get_scalar_exceptionä s(
|
ÿ
|
ÿ
|
ÿ
|
þ
|
ýz+TestBaseArithmeticOps._get_scalar_exceptionc C s d }| ||¡}|dkrVtj |¡s<tj |¡s<tj |¡rVtsVtjj d| d}n¶|rxtjj t
|
| d| dd}n|dkr°tj |¡stj |¡r°ts°tjj tj dd}n\|d krÜtj |¡rÜtsÜtjj tj dd}n0|d
|
kr tj |¡r tr tjj tj d d}|S ) NÚ__rpow__z<GH#29997: 1**pandas.NA == 1 while 1**pyarrow.NA == NULL for rm z not supported betweenpd.NA and z Python scalarr Ú __rfloordiv__z divide by 0Ú __rtruediv__Ú__pow__z'Invalid decimal function: power_checked) rF rE rF rH rg rK r rq rr rs rÅ r )rw rÈ rU rr rI r# r# r$ Ú_get_arith_xfail_markerû sj ÿ
|
ý
|
ü
|
ûù þýÿ
|
þ
|
þýþÿ
|
þýþÿ
|
þýþz-TestBaseArithmeticOps._get_arith_xfail_markerc s® |j j}|dkr2tj |¡s(tj |¡r2t d¡ | ||¡| _ |
|
||¡}|d k r`|j |¡ |dkrttj |¡stj |¡stj |¡r| td| j¡ t ||¡ d S ©NrG z%Skip testing Python string formattingÚ __floordiv__r; )r% r rE rF rP rQ rq rª rK Zseries_scalar_excrP ro rp rg rN rM Úsetattrr) r? rt Ú test_arith_series_with_scalar©rw rV Úall_arithmetic_operatorsr" Ú monkeypatchrU rr rx r# r$ rT 5 s0
|
ÿ
|
ÿ
|
ÿ þ
|
ý
|
û
|
ú
|
z3TestBaseArithmeticOps.test_arith_series_with_scalarc s® |j j}|dkr2tj |¡s(tj |¡r2t d¡ | ||¡| _ |
|
||¡}|d k r`|j |¡ |dkrttj |¡stj |¡stj |¡r| td| j¡ t ||¡ d S rQ )r% r rE rF rP rQ rq rª rK Zframe_scalar_excrP ro rp rg rN rM rS r) r? rt Ú test_arith_frame_with_scalarrU rx r# r$ rX T s0
|
ÿ
|
ÿ
|
ÿ þ
|
ý
|
û
|
ú
|
z2TestBaseArithmeticOps.test_arith_frame_with_scalarc
|
C s |j j}| ||¡| _|dkrNtj |¡rNtsN|j t
|
j j tj d| d¡ | ||¡}|d k rn|j |¡ |}t |¡}t tj|jd gt| |j d¡} tj |¡sÚtj |¡rÂ|dksÚtj |¡sÚtj |¡rê| td| j¡ | j||| | jd d S ) NrC zJImplemented pyarrow.compute.subtract_checked which raises on overflow for r r rD )Ú __truediv__rN r; )rJ ) r% r rK Zseries_array_excrE rF rJ r ro rp rq rr rs r rP rS r« rT r* r rH rg rN rM rS r) r? Z check_opname)
|
rw rV rV r" rW rU rr r¦ r¥ r= r# r# r$ Ú test_arith_series_with_arrays sJ ÿÿÿ
|
úù ýÿ
|
|
&
|
ÿ
|
ýü
|
ú
|
ù z2TestBaseArithmeticOps.test_arith_series_with_arrayc sÔ |j j}tj |¡r^tj |¡s^t |¡}tr4d}nd}t j
|
t |d || W 5 Q R X d S t rntj |¡sztj |¡r|j t jjt d| d¡ n*| d¡rÄ|j t jjtjd| d¡ t |¡ d S )Nz7Function add_checked has no kernel matching input typesú9Function 'add_checked' has no kernel matching input typesr z add_checked not implemented for r Zint8zraises on overflow for )r% r rE rF r± rN rS r« r rq r rR r rG ro rp rr rs r r rt Ú$test_add_series_with_extension_array)rw rV r" rU r¥ rÉ rx r# r$ r\ ¥ s4
|
|
ÿþÿ
|
þÿz:TestBaseArithmeticOps.test_add_series_with_extension_array)rz r{ r| rR Z
|
divmod_excÚ classmethodr+ r6 r? rF rK rP rT rX rZ r\ r} r# r# rx r$ r) s
|
|
&: 2r) c @ s8 e Zd Zdd Zdd Zej ddddg¡d d
|
Zd S ) ÚTestBaseComparisonOpsc
|
|
C s t |¡}t t j|jd gt| |jd¡}|jdkr|||}|d |ksTt|d |ksdt| ||¡}||d< ||d< | ||¡ nd }z|||}W n& t
|
k
|
rÆ } z| }W 5 d } ~ X Y nX |d krê| ||¡}| ||¡ n&t t |¡ | ||¡ W 5 Q R X d S )Nr rD )ÚeqÚneé éa )rS r« rT r* r r% rz r r# r¤ Ú Exceptionrq r rX )
|
rw rV Ú comparison_oprk r¥ r= r r§ rJ Úerrr# r# r$ Útest_compare_arrayÌ s(
|
&
|
|
z(TestBaseComparisonOps.test_compare_arrayc C s* t jtdd ||t W 5 Q R X d S )Nz'.* not implemented for <class 'object'>r )rq r rR Úobject)rw rV rd r# r# r$ Útest_invalid_other_compì s
|
ÿz-TestBaseComparisonOps.test_invalid_other_compÚ masked_dtypeÚbooleanZInt64r c C s ddd g}t j||d}t j|| ¡ dd}|||}|tjtjtjfkrZddd g}n
|
ddd g}t j|tt ¡ d}t
|
||¡ d S )Nr+ r rD rÎ FT) rS r« ÚlowerÚoperatorÚltÚgtr` r rE Úbool_r r¤ ) rw ri rd rV Ú
|
ser_maskedÚser_par Úexpr§ r# r# r$ Útest_comp_masked_numpyó s
|
|
|
z,TestBaseComparisonOps.test_comp_masked_numpyN) rz r{ r| rf rh rq rr rµ rs r# r# r# r$ r^ Ë s r^ c @ s e Zd ZdZdd Zej dddddgfej dddgfddddgfe
|
d¡dddgfddddgfe
|
d¡dddgfg¡dd Z d
|
d Z ej dddddgfej dddgfddddgfddddgfe
|
d¡dddgfe
|
d¡dddgfg¡d d Zdd Zej dddddgfej dddgfddddgfe
|
d¡dddgfe
|
d¡dddgfg¡dd Zej dddgddgddgg¡dd ZdS )ÚTestLogicalOpsz1Various Series and DataFrame logical ops methods.c
|
C sÜ t jdgd dgd d gd dd}t jddd gd dd}||B }t jdddddd dd d g dd}t ||¡ ||B }t ||¡ t |t jdgd dgd d gd dd¡ t |t jddd gd dd¡ d S ©NTrÝ Fúboolean[pyarrow]rD ©rS r« r r¤ ©rw r9 r: r r§ r# r# r$ Útest_kleene_or s$ (þ &þ ÿz TestLogicalOps.test_kleene_orzother, expectedNTFc C sj t jddd gdd}||B }t j|dd}t ||¡ ||B }t ||¡ t |t jddd gdd¡ d S ©NTFrv rD rw ©rw r= r§ r9 r r# r# r$ Útest_kleene_or_scalar s ÿz$TestLogicalOps.test_kleene_or_scalarc
|
C sÜ t jdgd dgd d gd dd}t jddd gd dd}||@ }t jddd dddd dd g dd}t ||¡ ||@ }t ||¡ t |t jdgd dgd d gd dd¡ t |t jddd gd dd¡ d S ru rw rx r# r# r$ Útest_kleene_and4 s$ (þ &þ ÿz TestLogicalOps.test_kleene_andc C sj t jddd gdd}||@ }t j|dd}t ||¡ ||@ }t ||¡ t |t jddd gdd¡ d S rz rw r{ r# r# r$ Útest_kleene_and_scalarJ s ÿz%TestLogicalOps.test_kleene_and_scalarc
|
C sÜ t jdgd dgd d gd dd}t jddd gd dd}||A }t jddd ddd d d d g dd}t ||¡ ||A }t ||¡ t |t jdgd dgd d gd dd¡ t |t jddd gd dd¡ d S ru rw rx r# r# r$ Útest_kleene_xorc s$ (þ &þ ÿz TestLogicalOps.test_kleene_xorc C sj t jddd gdd}||A }t j|dd}t ||¡ ||A }t ||¡ t |t jddd gdd¡ d S rz rw r{ r# r# r$ Útest_kleene_xor_scalary s ÿz%TestLogicalOps.test_kleene_xor_scalarzop, expÚ__and__Ú__or__Ú__xor__c C s` ddd g}t j|dd}t j|dd}t|||}t j|dd gtt ¡ d}t ||¡ d S )NTFrj rD rv )rS r« r¡ r rE ro r r¤ )rw r> rr rV rp rq r r§ r# r# r$ Útest_logical_masked_numpy s
|
|
z(TestLogicalOps.test_logical_masked_numpy)rz r{ r| Ú__doc__ry rq rr rµ rS rj r ro r| r} r~ r r r r# r# r# r$ rt sX úþ
|
úþ
|
ûþ
|
|
ýþrt c C sr t jtdd t d¡ W 5 Q R X t d¡} tt dd¡}| |ksJtt jtdd t d¡ W 5 Q R X d S )NzPassing pyarrow typer z#not_a_real_dype[s, tz=UTC][pyarrow]z timestamp[s, tz=UTC][pyarrow]r7 ÚUTCzdecimal(7, 2)[pyarrow])rq r rR r rå rE Ú timestampr )r% r§ r# r# r$ ÚFtest_arrowdtype_construct_from_string_type_with_unsupported_parameters£ s
|
r c C s8 d} d}t jt|d tjtd| d W 5 Q R X d S )Nz int64[pyarrow]foobar[pyarrow]zYPassing pyarrow type specific parameters \(\[pyarrow\]\) in the string is not supported\.r rÝ rD )rq r rR rS r« Úrange)ÚinvalidrÉ r# r# r$ Ú;test_arrowdtype_construct_from_string_type_only_one_pyarrow° s
|
ÿr Ú interpolationZlinearrk ÚhigherZnearestZmidpointÚquantiler* c C sN | j j}| dddg¡} t | ¡}tj |¡sFtj |¡sFtj |¡rt
|
rPd}nd}t j tj |d |j||d W 5 Q R X d S tj |¡sÞtj |¡sÞtj |¡r¬t
|
s¬n2tj | jj¡r¾n |j t jjtj d| d¡ | dddg¡} t | ¡}|j||d}tj |¡rÆ|dkrÆ|jr>d |j d
|
|j d }nd |j d }| |¡} | j||d}
|
|d kr|jd kr|
|
j dd}
|
||
|
ksÂt n,|jd kr®|
|
j d ¡}
|
t! "||
|
| j ¡¡ d S |d krä|| d ksJt nftj| ddg¡d d gd}
|
tj |¡s*tj |¡s*tj |¡r>|
|
d¡}
|
| d¡}t! "||
|
¡ d S )Nr z;Function quantile has no kernel matching input types \(.*\)z=Function 'quantile' has no kernel matching input types \(.*\)r )Úqr z&quantile not supported by pyarrow for r )rk r zM8[z, r9 r* r F)Úwarn)r úfloat64[pyarrow])#r% r ÚtakerS r« rE rF rP rQ rG r rq r r r rg rH rK r± r rX ro rp rr rs rM r rì r£ Ú to_pydatetimer ÚdtÚfloorr r¤ ) rV r r r" rU r¥ rÉ r Zpd_dtypeZser_npr§ r# r# r$ Ú test_quantile» sz
|
|
ÿ
|
þ
|
ý
|
ÿ
|
þ
|
ýýþÿ
|
|
|
|
|
ÿ
|
þ
|
ý
|
|
r ztake_idx, exp_idxr/ r& Z
|
multi_modeZ single_mode)r c C s@ | |¡}t |¡}|jdd}t | |¡¡}t ||¡ d S )NT©Zdropna)r rS r« Úmoder r¤ )ra Ztake_idxZexp_idxrV r¥ r r§ r# r# r$ Útest_mode_dropna_true s
|
|
|
r c C sx t jd d | d g| jd}|jdd}t jd g| jd}t ||¡ t jd | d g| jd}|jdd}t ||¡ d S )Nr rD Fr )rS r« r% r r r¤ )rV Z more_nansr r§ r# r# r$ Ú test_mode_dropna_false_mode_na s r zarrow_dtype, expected_typeé c C s t | j|kstd S r )r rX r )Z arrow_dtypeÚ expected_typer# r# r$ Útest_arrow_dtype_type$ s r c C sj t t dddg¡} t| s ttjj | ¡s2tt t
|
t | ¡}|| }|t | ¡ }t ||¡ d S )NTF)r rE rT r r rS ÚcoreÚcommonZis_bool_indexerr« r r r Zasarrayr r¤ )rV r7 r r§ r# r# r$ Útest_is_bool_dtype7 s r c C sJ | j j}tj |¡s,tj |¡s,tj |¡r:t| sFtn t| rFtd S r ) r% r rE rF rH rg rK r r ©rV r r# r# r$ Útest_is_numeric_dtypeB s
|
ÿ
|
þ
|
ýr¢ c C s2 | j j}tj |¡r"t| s.tn t| r.td S r )r% r rE rF rg r r r¡ r# r# r$ Útest_is_integer_dtypeO s r£ c C s2 | j j}tj |¡r"t| s.tn t| r.td S r )r% r rE rF rg r r r¡ r# r# r$ Útest_is_any_integer_dtypeX s r¤ c C s2 | j j}tj |¡r"t| s.tn t| r.td S r )r% r rE rF rI r r r¡ r# r# r$ Ú test_is_signed_integer_dtypea s r¥ c C s2 | j j}tj |¡r"t| s.tn t| r.td S r )r% r rE rF rJ r r r¡ r# r# r$ Ú test_is_unsigned_integer_dtypei s r¦ c C s2 | j j}tj |¡r"t| s.tn t| r.td S r )r% r rE rF rH r r r¡ r# r# r$ Útest_is_float_dtypeq s r§ c C sl t | ¡}| d¡}t |¡}t |¡}t|t|ks<tt |¡}t ||¡ t |¡}t ||¡ d S )Nr/ )
|
rS r« ÚheadÚpickleÚdumpsr r Úloadsr r¤ )rV r§ Zexpected_slicedZ full_pickledZsliced_pickledr Z result_slicedr# r# r$ Útest_pickle_roundtripy s
|
|
|
|
|
|
r¬ c C sH | j ¡ j}| | j¡}t|jtr(tt|jts8tt || ¡ d S r )
|
r Ú to_pandasrT r£ r% r
r r r r )rV Zpd_arrayr r# r# r$ Ú test_astype_from_non_pyarrow s
|
r® c C s6 t dg¡} | d¡}t jdgdd}t ||¡ d S )Nz1.0r r( rD )rS r« r£ r r¤ ©r¥ r r§ r# r# r$ Ú&test_astype_float_from_non_pyarrow_str s
|
r° c C st | ¡ }| jj}tj |¡s(tj |¡r8t t | ¡}n t | j¡}| j
|
rd| t ¡}t j|t | ¡< t ||¡ d S r )Úto_numpyr rX rE rF rN rM r rT ÚlistZ_hasnar£ rg rS rj r÷ r Úassert_numpy_array_equal)rV r r r§ r# r# r$ Útest_to_numpy_with_defaults s
|
r´ c C sT dd g} t j| dd}| ¡ }tjdt jgtd}t|d tsDtt
|
||¡ d S )Nr+ r rD r ) rS rT r± r rj rg r
Úintr r r³ )rV Úarrr r§ r# r# r$ Útest_to_numpy_int_with_na« s r· z na_val, exp)r+ r+ c C sH t jt jt jgdd}|jd| d}tj|gd dd}t ||¡ d S )Nú null[pyarrow]rD rÍ )r% rk r/ ©rS rT rj r± r r r³ )Zna_valrr r¶ r r§ r# r# r$ Útest_to_numpy_null_arrayµ s rº c C sH t jt jt jgdd} | jd d}tjt jgd dd}t ||¡ d S )Nr¸ rD r/ rg r¹ )r¶ r r§ r# r# r$ Ú!test_to_numpy_null_array_no_dtype¾ s r» c C s´ | ¡ }| ¡ }| d |d d
< ttj| d gt| | jjd}t ||¡ | ¡ }| d d d
|d d
< | d d d
}t ||¡ | ¡ }| ¡ |d d
< | }t ||¡ d S )Nr ©rX r- )
|
Úcopyr rE rT r r rX r r Útolist)rV Úorigr r§ r# r# r$ Útest_setitem_null_sliceÆ s ÿ rÀ c C s | j j}tj |¡s tj |¡r.d}t}d}n@tj |¡sRtj |¡sRtj |¡rbd}tj
|
}d}n d}t}d}t j ||d || d d
< W 5 Q R X d S )Né{ z Invalid value '123' for dtypeÚfoozCould not convertz Invalid value 'foo' for dtyper ) r rX rE rF rP rQ rÅ rg rH rG r rq r )rV r Z
|
fill_valuere rÉ r# r# r$ Útest_setitem_invalid_dtypeÜ s&
|
ÿ
|
þ
|
ýrà zreturns object with 7.0rm c C sv t jt d¡t d¡gt ¡ d} | jt ¡ tt ¡ ijd}tj t d¡t d¡gtt ¡ d}t
|
||¡ d S )Nz
|
2019-12-31r¼ ©Z types_mapperrD ) rE rT rS Ú TimestampÚdate32r r Zdate64Úgetr« r r¤ )Z
|
date_arrayr r§ r# r# r$ Ú&test_from_arrow_respecting_given_dtypeó s ÿÿ
|
þrÈ zdoesn't raise with 7c C sT t jddgt ¡ d} tjt jdd$ | jt ¡ tt ¡ ij d W 5 Q R X d S )NrÏ ç @r¼ z Float value 1.5 was truncatedr rÄ )
|
rE rT rÍ rq r r r r rÌ rÇ )rT r# r# r$ Ú-test_from_arrow_respecting_given_dtype_unsafe s rÊ c C s d} t jdddt jg| d}| d¡}t jdddt jg| d}t ||¡ t jd t jd
|
g| d}| d ¡}t jd t jd g| d}t ||¡ d S )Nr r) g®Gáz®ó?g{®Gáz@rD r+ g333333ó?gÍÌÌÌÌÌ@gÙ^@g¤p=
|
×cL@r- g ^@g N@)rS r« rj Úroundr r¤ )r% r¥ r r§ r# r# r$ Ú
|
test_round s
|
|
rÌ c C s^ | \}}}| dddg¡}tj|d< |r2t |¡}d}tjt|d | |¡ W 5 Q R X d S )Nr/ r r+ r- zOsearchsorted requires array to be sorted, which is impossible with NAs present.r )r rS rj r« rq r rÛ Z searchsorted)rd r' r: r\ r9 r¶ rÉ r# r# r$ Ú test_searchsorted_with_na_raises s
|
|
|
ÿrÍ ÚpatÚabcz a[a-z]{2}c C sP t jdd gtt ¡ d}|j | ¡}t jdd gtt ¡ d}t ||¡ d S )NrÏ rD r+ )
|
rS r« r rE r r r· Úint32r r¤ )rÎ r¥ r r§ r# r# r$ Útest_str_count( s rÑ c C sH t jdd gtt ¡ d} tjtdd | jj ddd W 5 Q R X d S )NrÏ rD z count notr r+ ©Úflags)
|
rS r« r rE r rq r rR r r· ©r¥ r# r# r$ Ú test_str_count_flags_unsupported0 s rÕ zside, str_funcr- Úrjustr. ÚljustZbothÚcenterc C sb t jdd gtt ¡ d}|jjd| dd}t jtd|ddd gtt ¡ d}t ||¡ d S )Nr9 rD rÝ r3 )ÚwidthÚsideZfillchar)
|
rS r« r rE r r Úpadr¡ r r¤ )rÚ Zstr_funcr¥ r r§ r# r# r$ Ú test_str_pad6 s
|
ÿrÜ c C sH t jdd gtt ¡ d} tjtdd | j ddd¡ W 5 Q R X d S )Nr9 rD zInvalid side: foor rÝ r r3 )
|
rS r« r rE r rq r rÛ r rÛ rÔ r# r# r$ Útest_str_pad_invalid_sideB s rÝ zpat, case, na, regex, expÚabFTZAbz a[a-z]{1}z A[a-z]{1}c C sT t jdd gtt ¡ d}|jj| |||d}t j|tt ¡ d}t ||¡ d S )NrÏ rD )ÚcaseÚnaÚregex)
|
rS r« r rE r r Úcontainsro r r¤ )rÎ rß rà rá rr r¥ r r§ r# r# r$ Útest_str_containsH s rã c C sH t jdd gtt ¡ d} tjtdd | jj ddd W 5 Q R X d S )NrÏ rD z contains notr r9 r+ rÒ )
|
rS r« r rE r rq r rR r râ rÔ r# r# r$ Ú#test_str_contains_flags_unsupportedY s rä zside, pat, na, expÚ
|
startswithr: ÚendswithZbcc C sT t jdd gtt ¡ d}t|j| ||d}t j|tt ¡ d}t ||¡ d S )NrÏ rD )rà ©
|
rS r« r rE r r¡ r ro r r¤ )rÚ rÎ rà rr r¥ r r§ r# r# r$ Útest_str_start_ends_with_ s
|
rè z arg_name, argÚreplrß rÓ r+ c C sZ t jdd gtt ¡ d}dddd}||| < tjtdd |jj f | W 5 Q R X d S ) NrÏ rD r: r3 T)rÎ ré rá zreplace is not supportedr )
|
rS r« r rE r rq r rR r Úreplace)Zarg_nameÚargr¥ r/ r# r# r$ Ú test_str_replace_unsupportedo s
|
rì zpat, repl, n, regex, expr9 r3 r- ZxbxcZxbacz[a-b]Zxxxcc C sT t jdd gtt ¡ d}|jj| |||d}t j|tt ¡ d}t ||¡ d S )NZabacrD )Únrá ) rS r« r rE r r rê r r¤ )rÎ ré rí rá rr r¥ r r§ r# r# r$ Útest_str_replace{ s rî c C sH t jdd gtt ¡ d} tjtdd | j ddg¡ W 5 Q R X d S )NrÏ rD z repeat is notr r+ r/ )
|
rS r« r rE r rq r rR r ÚrepeatrÔ r# r# r$ Útest_str_repeat_unsupported s rð zUnsupported for pyarrow < 7r° c C sP t jdd gtt ¡ d} | j d¡}t jdd gtt ¡ d}t ||¡ d S )NrÏ rD r/ Zabcabc) rS r« r rE r r rï r r¤ r¯ r# r# r$ Útest_str_repeat s rñ zpat, case, na, expc C sR t jdd gtt ¡ d}|jj| ||d}t j|tt ¡ d}t ||¡ d S ©NrÏ rD )rß rà ©
|
rS r« r rE r r r ro r r¤ ©rÎ rß rà rr r¥ r r§ r# r# r$ Útest_str_match s rõ ZAbcc C sR t jdd gtt ¡ d}|jj| ||d}t j|tt ¡ d}t ||¡ d S rò ró rô r# r# r$ Útest_str_fullmatch® s rö z sub, start, end, exp, exp_typrÝ c C sN t jdd gtt ¡ d}|jj| ||d}t j|t|d}t ||¡ d S )NrÏ rD )ÚstartÚend) rS r« r rE r r Úfindr r¤ )Úsubr÷ rø rr Zexp_typr¥ r r§ r# r# r$ Ú test_str_findÀ s rû c C sH t jdd gtt ¡ d} tjtdd | jj ddd W 5 Q R X d S )NrÏ rD zfind not implementedr rÞ r+ )r÷ )
|
rS r« r rE r rq r rR r rù rÔ r# r# r$ Ú test_str_find_notimplementedË s rü zi, expÚer\ éýÿÿÿc C sN t jddd gtt ¡ d}|j | ¡}t j|tt ¡ d}t ||¡ d S )NrÏ ÚderD ) rS r« r rE r r rÇ r r¤ )Úirr r¥ r r§ r# r# r$ Ú test_str_getÑ s r z=TODO: StringMethods._validate should support Arrow list typesc C sZ t tt tdtdd g¡¡} | j d¡}t jddd gtt ¡ d}t
|
||¡ d S )NrÏ Ú123ú=za=b=cz1=2=3rD ) rS r« r rE rT r² r Újoinr r r r¤ r¯ r# r# r$ Ú test_str_joinâ s " r zstart, stop, step, expc C sP t jdd gtt ¡ d}|j | ||¡}t j|tt ¡ d}t ||¡ d S ©NÚabcdrD ) rS r« r rE r r Úslicer r¤ )r÷ ÚstopÚsteprr r¥ r r§ r# r# r$ Útest_str_sliceí s r zstart, stop, repl, expZaxcdZxcdZcdc C sP t jdd gtt ¡ d}|j | ||¡}t j|tt ¡ d}t ||¡ d S r ) rS r« r rE r r Z slice_replacer r¤ )r÷ r ré rr r¥ r r§ r# r# r$ Útest_str_slice_replaceü s r zvalue, method, expZa1cÚisalnumz!|,ZaaaÚisalphaz!!!u Ù Ú isdecimalz~!r; Úisdigitú~ÚislowerZaaAr Ú isnumericZ11Iú ÚisspaceÚ zThe ThatÚistitlezthe ThatZAAAÚisupperZAAcc C sR t j| d gtt ¡ d}t|j| }t j|d gtt ¡ d}t ||¡ d S )NrD rç )Úvaluerö rr r¥ r r§ r# r# r$ Útest_str_is_functions s r z method, expÚ
|
capitalizezAbc defÚtitlezAbc DefÚswapcasezAbC Defzabc defÚupperzABC DEFÚcasefoldc C sR t jdd gtt ¡ d}t|j| }t j|d gtt ¡ d}t ||¡ d S )NzaBc dEFrD © rS r« r rE r r¡ r r r¤ )rö rr r¥ r r§ r# r# r$ Ú test_str_transform_functions) s r! c C sN t jdd gtt ¡ d} | j ¡ }t jdd gtt ¡ d}t ||¡ d S )Nr rD r& )
|
rS r« r rE r r r rÐ r r¤ r¯ r# r# r$ Ú test_str_len; s
|
r" zmethod, to_strip, valr5 z abc ZxabcxÚlstripz abcZxabcÚrstripzabc Zabcxc C sV t j|d gtt ¡ d}t|j| |d}t jdd gtt ¡ d}t ||¡ d S )NrD )Úto_striprÏ r )rö r% Úvalr¥ r r§ r# r# r$ Útest_str_stripB s r' r& Zabc123c C sP t j| d gtt ¡ d}|j d¡}t jdd gtt ¡ d}t ||¡ d S ©NrD r rÏ ) rS r« r rE r r Z removesuffixr r¤ ©r& r¥ r r§ r# r# r$ Útest_str_removesuffixT s r* Z123abcc C sP t j| d gtt ¡ d}|j d¡}t jdd gtt ¡ d}t ||¡ d S r( ) rS r« r rE r r Z removeprefixr r¤ r) r# r# r$ Útest_str_removeprefix\ s r+ ÚerrorsÚignoreÚstrictz encoding, expÚutf8s abcÚutf32s ÿþ a b c c C sR t jdd gtt ¡ d}|j || ¡}t j|d gtt ¡ d}t ||¡ d S ©NrÏ rD )
|
rS r« r rE r r ÚencodeÚbinaryr r¤ )r, Úencodingrr r¥ r r§ r# r# r$ Útest_str_encoded s r5 c C s` t jddd gtt ¡ d}|jjd| d}t jdgg d gtt t ¡ ¡d}t ||¡ d S )NrÏ ZefgrD r: rÒ )
|
rS r« r rE r r ÚfindallÚlist_r r¤ )rÓ r¥ r r§ r# r# r$ Útest_str_findalls s $r8 rö r Úrindexz
|
start, endc C s t jdd gtt ¡ d}t|j| d||}t jdd gtt ¡ d}t ||¡ t
|
j t dd t|j| d|| W 5 Q R X d S )NÚabcbarD r\ r/ zsubstring not foundr r ) rS r« r rE r r¡ r rÌ r r¤ rq r rÛ )rö r÷ rø r¥ r r§ r# r# r$ Útest_str_r_index{ s r; ÚformÚNFCÚNFKCc C s> t jdd gtt ¡ d}|j | ¡}| ¡ }t ||¡ d S r1 )
|
rS r« r rE r r r r½ r r¤ )r< r¥ r r§ r# r# r$ Útest_str_normalize s r? c C sX t jddd gtt ¡ d}|j d| |¡}t jddd gtt ¡ d}t ||¡ d S )Nr: r rD r\ r/ r- )
|
rS r« r rE r r ÚrfindrÌ r r¤ )r÷ rø r¥ r r§ r# r# r$ Útest_str_rfind s rA c C sT t jdd gtt ¡ d} | j ddi¡}t jdd gtt ¡ d}t ||¡ d S )Nr: rD rb r: Zbbcbb) rS r« r rE r r Ú translater r¤ r¯ r# r# r$ Útest_str_translate£ s rC c C sP t jdd gtt ¡ d} | j d¡}t jdd gtt ¡ d}t ||¡ d S )Nr: rD rÝ zabc
|
ba) rS r« r rE r r Úwrapr r¤ r¯ r# r# r$ Ú test_str_wrapª s rE c C sl t jdd dgtt ¡ d} | j ¡ }t jdddgdddgdddggtt ¡ dddgd }t
|
||¡ d S )
|
Nza|bza|crD TFr9 r: r\ )r% rà ) rS r« r rE r r Z get_dummiesrÞ ro r r r¯ r# r# r$ Útest_get_dummies± s
|
|
ýrF c C s t jdd gtt ¡ d} | j d¡}t jdddgd d d ggtt ¡ d}t ||¡ | jjddd}t t
|
t dddgd g¡¡}t ||¡ | j d¡}t jdddgd d d ggtt ¡ d}t ||¡ | jj ddd}t t
|
t dddgd g¡¡}t ||¡ d S ) Nr: rD r: r9 ZcbaF)ÚexpandrÏ )rS r« r rE r r Ú partitionrÞ r r r rT r¤ Ú
|
rpartitionr¯ r# r# r$ Útest_str_partition¼ s&
|
ÿ
|
ÿ rJ c C s( t jddd gtt ¡ d} | j d¡}t tt dddgdddgd g¡¡}t
|
||¡ | jjddd }t tt dd
|
gdd
|
gd g¡¡}t
|
||¡ | jjd d d }t tt ddgddgd g¡¡}t
|
||¡ | jjd d d d}t tt ddd g¡tt ddd g¡d¡}t ||¡ d S )NÚa1cbcbÚa2cbcbrD r\ Úa1r: Úa2r+ ©rí Zbcbz[1-2]T)rá r9 Zcbcb)rá rG ©r r+ ) rS r« r rE r r Úsplitr rT r r¤ rÞ r r¯ r# r# r$ Útest_str_splitÓ s. ÿ ÿ ÿ þÿrR c C sê t jddd gtt ¡ d} | j d¡}t tt dddgdddgd g¡¡}t
|
||¡ | jjddd }t tt d
|
dgd dgd g¡¡}t
|
||¡ | jjddd d }t tt d
|
d d g¡tt ddd g¡d¡}t ||¡ d S )NrK rL rD r\ rM r: rN r+ rO Za1cbZa2cbT)rí rG rP ) rS r« r rE r r Úrsplitr rT r r¤ rÞ r r¯ r# r# r$ Útest_str_rsplitò s$ ÿ ÿ þÿrT c C sD t jdd gtt ¡ d} tjtdd | j d¡ W 5 Q R X d S )NrÏ rD z,str.extract not supported with pd.ArrowDtyper z[ab](\d))
|
rS r« r rE r rq r rR r ÚextractrÔ r# r# r$ Ú test_str_unsupported_extract s ÿrV rì rë r r8 r7 c C sF ddg}t | ¡}tj||d}tt jdd g|d}t ||¡ d S )NZ1000ÚNaTrD iè r¼ )rE Údurationr r rT r r )rì Ústringsr r r§ r# r# r$ Ú#test_duration_from_strings_with_nat s
|
|
rZ c C s> | j j}tj |¡s:tjtdd t | ¡j
|
W 5 Q R X d S )Nz2Can only use .dt accessor with datetimelike valuesr ) r% r rE rF r± rq r ÚAttributeErrorrS r« r rh r# r# r$ Útest_unsupported_dt s ÿr\ zprop, expectedÚyearéç ÚdayZ day_of_weekZ dayofweekÚweekdayZ day_of_yearZ dayofyearÚhourÚminuteZ is_leap_yearz.is_leap_year not implemented for pyarrow < 8.0r )ZmarksÚ microsecondr8 ÚmonthÚ
|
nanosecondé ZquarterÚsecondrb r r c C s t jt jddddddddd d gtt d
|
¡d }t|j| }d }t|t rVt
|
¡ }nt|t rjt d
|
¡}t t tj|d g|d ¡}t ||¡ d S ) Nr^ r+ r/ rÝ r& rb r8 rf )r] rd r_ ra rb rg rc re rë rD r¼ )rS r« rÅ r rE r r¡ r r
r rÆ r Útime64r rT r r¤ )Úpropr§ r¥ r Zexp_typer# r# r$ Útest_dt_properties& s. ø
|
õ ò
|
|
|
|
rj c C sh t jtdddddd gtt | ¡d}|jj}t ttj tddd gt
|
| ¡d¡}t ||¡ d S ) Nr^ r+ r/ rÝ ©r] rd r_ ra rD r r¼ ) rS r« r r rE r r r r rT rh r r¤ )rì r¥ r r§ r# r# r$ Útest_dt_time_preserve_unit_ s þ ÿrl r r ú
|
US/Pacificc C sD t jtdddddd gttjd| dd}|jj}|| ks@td S ) Nr^ r+ r/ rÝ rk rë ©r rD ) rS r« r r rE r r r r )r r¥ r r# r# r$ Ú
|
test_dt_tzl s þro c C sh t jtdddddd gtt d¡d} | j ¡ }t jdddgdddggd d
|
d gd d }t
|
||¡ d S )Nr^ r+ r/ rÝ rk rë rD r r] Úweekr_ r )rà r% ) rS r« r r rE r r Ú isocalendarrÞ r r r¯ r# r# r$ Útest_dt_isocalendarv s þ
|
ýrr c C s t r&t r&| j tjjtjdd¡ t j
|
t dddddd gt t d¡d }|j d
|
¡}t j
|
d d gt t ¡ d }t ||¡ d S ) Nr r r^ r+ r/ rÝ rk rë rD z%Y-%m-%dT%H:%M:%Sz 2023-01-02T03:00:00.000000000)r r ro rp rq rr rs rE r rS r« r r r r Ústrftimer r r¤ )r" r¥ r r§ r# r# r$ Útest_dt_strftime s" ýÿ þ
|
ÿrt Úceilr rË c C s t jtdddddd gtt d¡d}tjtdd t |j
|
| d
|
d d W 5 Q R X tjtd d t |j
|
| d
|
d d W 5 Q R X d S )Nr^ r+ r/ rÝ rk rë rD zambiguous is not supported.r Z1HrW ©Z ambiguousz nonexistent is not supported.©Ú nonexistent) rS r« r r rE r rq r rR r¡ r ©rö r¥ r# r# r$ Ú*test_dt_roundlike_tz_options_not_supported s þ rz c C s t jtdddddd gtt d¡d}tjtdd t |j
|
| d
|
W 5 Q R X tjtd d t |j
|
| d W 5 Q R X d S ) Nr^ r+ r/ rÝ rk rë rD zfreq='1B' is not supportedr Z1Bz$Must specify a valid frequency: None) rS r« r r rE r rq r rÛ r¡ r ry r# r# r$ Ú"test_dt_roundlike_unsupported_freq§ s þr{ z'Methods not supported for pyarrow < 7.0ÚfreqÚDÚHÚTÚSÚLÚUÚNc C sn t tddddd g¡}tt d¡}t|j|d| |¡}t| |¡j|d| }t
|
||¡ d S )Nr^ r+ )r] rd r_ rë r0 ) rS r« r r rE r r¡ r r£ r r¤ )r| rö r¥ rU r§ r r# r# r$ Útest_dt_ceil_year_floor´ s ÿ r c C s t dddt dddg} tj| tt d¡d}|j ¡ }tj | t
|
d}t ||¡ t dd |D sjt| d¡j ¡ }t ||¡ d S ) Nr2 r+ r^ rë rD c s s | ]}t |tkV qd S r )rX r )Ú.0Úresr# r# r$ Ú <genexpr>Ë s z(test_dt_to_pydatetime.<locals>.<genexpr>zdatetime64[ns])r rS r« r rE r r r r rT rg r r³ Úallr r£ )rV r¥ r r§ r# r# r$ Útest_dt_to_pydatetimeà s
|
r c C s t jtdddddd gtt d¡d} tjtdd | j j
|
d
|
d d W 5 Q R X tjtd d | j j
|
d
|
d d W 5 Q R X d S )Nr^ r+ r/ rÝ rk rë rD z ambiguous='NaT' is not supportedr r rW rv z"nonexistent='NaT' is not supportedrw ) rS r« r r rE r rq r rR r Ú tz_localizerÔ r# r# r$ Ú*test_dt_tz_localize_unsupported_tz_optionsÑ s þr c C sp t jtdddddd gttjdddd } | j d ¡}t jtdddddd gtt d¡d }t ||¡ d S )
|
Nr^ r+ r/ rÝ rk rë rm rn rD )
|
rS r« r r rE r r r r r¤ r¯ r# r# r$ Útest_dt_tz_localize_noneÝ s þ þr c C sª t r&t r&|j tjjtjdd¡ t j
|
t dddddd gt t | ¡d}|j d ¡}tjt dddddd gt | ¡d
|
}tj |d ¡}t
|
t|¡}t ||¡ d S ) Nr r r^ r+ r/ rÝ rk rD rm r¼ ©r r ro rp rq rr rs rE r rS r« r r r r r rT ZcomputeZassume_timezoner r r¤ )rì r" r¥ r Úexp_datar§ r# r# r$ Útest_dt_tz_localizeê s& ýÿ þ ÿr znonexistent, exp_dateZ shift_forwardr4 rk Zshift_backwardz 2023-03-12 01:59:59.999999999c C s¤ t r&t r&|j tjjtjdd¡ t j
|
t ddddddd gt t d ¡d
|
}|jjd | d }tj|d gt d ¡d }tj |d ¡}t
|
t|¡}t ||¡ d S )Nr r r^ rÝ r4 r/ é )r] rd r_ ra rb rë rD rm rw r¼ r )rx Zexp_dater" r¥ r r r§ r# r# r$ Ú test_dt_tz_localize_nonexistent
|
s ýÿ þr r c C sD t jd gdd}t|| |d}|r.| dk}nt j}||ks@td S )Nr rD r r )rS r« r¡ rj r )rÓ r r¥ r r§ r# r# r$ Ú#test_boolean_reduce_series_all_null
|
s
|
r r zstring[pyarrow]c C s> t d ¡ ¡}tj|| d}tjt|| d}t ||¡ d S )Nzthe quick brown foxrD )rE rT rQ rS r« r r r¤ )r% r¶ r¥ r§ r# r# r$ Ú test_series_from_string_array,
|
s r c C sT d} t t tj| ftjdg¡}| ¡ }d|tj| ftjd< |j|jksPt d S )Nih6 rD F)
|
r rE Z chunked_arrayr Zonesro r½ Úzerosr r )r r¶ r§ r# r# r$ Ú/test_setitem_boolean_replace_with_mask_segfault4
|
s
|
r r c C sh t jdddgt| d}| ¡ }t jddddddddgtt ¡ dd d
|
d d d ddgd}t ||¡ d S )Nr+ r/ rÝ rD rÏ g @rÉ r· r¾ rÀ Úminz25%z50%z75%Úmax)r% r )rS r« r ZdescriberE rÍ r r¤ )r rV r r§ r# r# r$ Útest_describe_numeric_data=
|
s
|
ýr r[ c C s t dd g¡} t dd g¡}t j| tt d¡d}t j|tt d¡d}|| }t jdd gtt d¡d}t ||¡ d S )Nr+ rë rD r/ )
|
rS Ú to_datetimeZ to_timedeltar« r rE r rX r r¤ )Zdata_tsZdata_tdZser_tsZser_tdr r§ r# r# r$ Ú2test_duration_overflow_from_ndarray_containing_natJ
|
s r )Ör
r r r r Údecimalr Úior r rl r© ÚreÚnumpyr rq Z pandas._libsr Z pandas.compatr
|
r r r r r r Z pandas.errorsr Zpandas.core.dtypes.commonr Zpandas.core.dtypes.dtypesr ZpandasrS Zpandas._testingZ_testingr Zpandas.api.typesr r r r r r r Zpandas.tests.extensionr Z importorskiprE Z pandas.core.arrays.arrow.arrayr Z pandas.core.arrays.arrow.dtyper ZfixtureZALL_PYARROW_DTYPESr r% rV rZ r[ ra rd re ri rk ZBaseCastingTestsrl ZBaseConstructorsTestsr~ ZBaseGetitemTestsr ZBaseAccumulateTestsr ZBaseNumericReduceTestsr¶ ZBaseBooleanReduceTestsrÑ ZBaseGroupbyTestsrÔ ZBaseDtypeTestsrã ZBaseIndexTestsrï ZBaseInterfaceTestsrð ZBaseMissingTestsrõ ZBasePrintingTestsrú ZBaseReshapingTestsrû ZBaseSetitemTestsrþ ZBaseParsingTestsr ZBaseUnaryOpsTestsr
|
ZBaseMethodsTestsr r: r) ZBaseComparisonOpsTestsr^ rt r r rr rµ r r r r3 ÚbytesZ large_binaryZ large_stringr7 rÌ r² Z
|
large_listZmap_r ÚdictÚ
|
dictionaryr r r¢ r£ r¤ r¥ r¦ r§ r¬ r® r° r´ r· Z
|
no_defaultr rº r» rÀ rà ZskipifrÈ rÊ rÌ rÍ rÑ rÕ rÜ rÝ rã rä rè Úcompilerì rî rð rs rR rñ rõ rö rÐ rû rü r r[ r r r r r! r" r' r* r+ r5 r8 r; r? rA rC rE rF rJ rR rT rV rZ r\ r! rj rl ro rr rt rz r{ r r r r r rÅ r r r r ZALL_INT_PYARROW_DTYPESZFLOAT_PYARROW_DTYPESr r r r# r# r# r$ Ú<module> sø $ $
|
|
;
|
|
|
:
|
|
|
|
FRb#4S' 37 " ÿ L,ý
|
|
|
|
øþ
|
|
|
|
|
|
ÿ
|
ûþ
|
|
üþ
|
þ
|
ýþ
|
ý
|
úþ
|
úþ
|
*þ
|
ûþ
|
|
þ
|
ýþ
|
ýþ
|
|
îþ
|
|
|
|
|
úþ úþ
|
|
þþþþ
|
þþ
|
ýý
|
|
|
çþ ÿ*
|
þþ ý
|
