U
|
®ý°d© ã @ sZ d dl mZmZmZ d dlmZ d dlZd dlZd dl Z d dl
|
m Z d dl m Z mZ d dlZd dlmZmZmZmZmZmZmZ d dlmZ d dlm Z m Z d dl m Z d d l!m"Z" e j#d
|
d d gd d gddd Z$dd Z%G dd dZ&G dd dZ'G dd dZ(G dd dZ)G dd d Z*G d d d Z+G d d d Z,d!d" Z-d#d$ Z.d%d& Z/dS )'é )ÚdateÚ timedeltaÚtimezone)ÚDecimalN)ÚIncompatibleFrequency)Úis_datetime64_dtypeÚis_datetime64tz_dtype)Ú CategoricalÚIndexÚSeriesÚ TimedeltaÚ bdate_rangeÚ
|
date_rangeÚisna)ÚnanopsÚops)Ú expressions)ÚNUMEXPR_INSTALLEDTi@B ÚnumexprÚpython)ZautouseÚparamsÚidsc c s t j}| jt _| jV |t _d S ©N)ÚexprÚ _MIN_ELEMENTSÚparam)Úrequestr © r úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/test_arithmetic.pyÚswitch_numexpr_min_elements& s r c C s | tj t| ¡¡S r )ÚtakeÚnpÚrandomÚ permutationÚlen)Úobjr r r Ú_permute. s r& c
|
@ s e Zd Zej ddd dd dfdd dd dfdd d d d
|
fd d d d d
|
fg¡ej d d dddddg¡dd Zdd Zdd Ze j
|
e j dfe j ejdfgZdD ]VZdZee eZee eZee d e Zefd dZe eeef¡ e eeef¡ q¨ej d e¡d d Zd!S )"ÚTestSeriesFlexArithmeticÚtsc C s | S r r ©Úxr r r Ú<lambda>6 ó z!TestSeriesFlexArithmetic.<lambda>c C s | d S ©Né r r) r r r r+ 6 r, Fc C s | S r r r) r r r r+ 7 r, c C s | d d d
S r- r r) r r r r+ 7 r, c C s | S r r r) r r r r+ 8 r, c C s dS )Né r r) r r r r+ 8 r, Tc C s t ¡ S r )ÚtmZmakeFloatSeriesr) r r r r+ 9 r, ÚopnameÚaddÚsubÚmulÚfloordivÚtruedivÚpowc C s t ¡ d¡}|d |}|d |}|d }tt|}tt|}|||} |||}
|
t | |
|
¡ |rttd| } | ||} |||}
|
t | |
|
¡ d S )Nr( r é r. Úr)r0 ZmakeTimeSeriesÚrenameÚgetattrr ÚoperatorÚassert_almost_equal) Úselfr1 r( ZtserÚseriesÚotherZ check_reverseÚopZaltÚresultÚexpectedÚropr r r Ú test_flex_method_equivalence3 s
|
|
|
|
|
|
z5TestSeriesFlexArithmetic.test_flex_method_equivalencec sV G fdddt |}tt |} dddgdd}d|_||d}|jdksRtd S ) Nc s" e Zd ZdgZe fddZdS )zZTestSeriesFlexArithmetic.test_flex_method_subclass_metadata_preservation.<locals>.MySeriesr* c s S r r )r> ©ÚMySeriesr r Ú _constructorX s zgTestSeriesFlexArithmetic.test_flex_method_subclass_metadata_preservation.<locals>.MySeries._constructorN)Ú__name__Ú
|
__module__Ú __qualname__Ú _metadataÚpropertyrH r rF r r rG U s rG r8 r. é Útest©Únameé* )r r; r* ÚAssertionError)r> Zall_arithmetic_operatorsr1 rA ÚmrB r rF r Ú/test_flex_method_subclass_metadata_preservationS s
|
|
zHTestSeriesFlexArithmetic.test_flex_method_subclass_metadata_preservationc C sD t ddtjdddg}| d¡ d¡}|jddd}t ||¡ d S )Nr r8 rN é r/ r. ©Ú
|
fill_value)r r! ÚnanÚfillnar2 r0 Úassert_series_equal)r> ÚserÚexpÚresr r r Ú test_flex_add_scalar_fill_valuec s z8TestSeriesFlexArithmetic.test_flex_add_scalar_fill_valuer8 )r2 r3 r4 r7 r6 r5 r r9 c C s t t||| S r )r; r< )r* ÚyrA r r r r+ r r, zop, equiv_op, fvc C s ddd}t tjdddtjgt d¡d}t tjd tjd
|
tjd gt d ¡d}|||}|||}t ||¡ ||||||d |||dd d S )Nr c
|
S s |j |j ¡}| |¡}| |¡}t|}t|}g }tt|D ]°} tjdd || r|| r|| tj ¡ W 5 Q R £ qB| |||| ¡ nV|| rÐ|| rº| tj ¡ W 5 Q R £ qB| ||| |¡ n| ||| || ¡ W 5 Q R X qB| |||d}
|
t
|
||} t |
|
| ¡ d S )NÚignore)ÚallrW ) ÚindexÚunionÚreindexr Úranger$ r! ÚerrstateÚappendrY r r0 r[ ) ÚmethrA ÚaÚbrX Ú exp_indexZamaskZbmaskÚ
|
exp_valuesÚirB rC r r r Ú _check_fillx s,
|
|
$
|
zDTestSeriesFlexArithmetic.test_operators_combine.<locals>._check_fillç ð?g @g @r/ ©rc r8 rN ç @é rW ©Úaxis)r )r r! rY Úaranger0 r[ ) r> rA Zequiv_opÚfvro rj rk rB r] r r r Útest_operators_combinev s
|
$
|
|
z/TestSeriesFlexArithmetic.test_operators_combineN)rI rJ rK ÚpytestÚmarkÚ parametrizerE rU r_ r Údivr< r6 Zrdivr ZrtruedivZpairingsrA rw r; ZlopZlequivrD Zrequivrh rx r r r r r' 2 s4 üþ ÿ
|
|
r' c @ sê e Zd Zdd Zej ddddgddgfd ddgd
|
d gfg¡d d Zdd Zej dde dg¡dd Z
|
dd Z dd Z dd Z dd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zej d+d,d-g¡d.d/ Zd0d1 ZdS )2ÚTestSeriesArithmeticc C s¶ t jdddd}ttj t|¡|d}||d d d
}|| }tj|jdd d
< t
|
||¡ |t |d d d
}t
|
||¡ d}t j t|d ||jd
|
d d W 5 Q R X d S ) Nú1/1/2000z1/1/2010ÚA©Úfreqrq r. r8 z4Input has different freq=D from Period\(freq=A-DEC\)©ÚmatchÚDÚend)Úhow)ÚpdZ period_ranger r! r" Úrandnr$ rY Úilocr0 r[ r& ry Úraisesr Zasfreq)r> Úrngr( rB rC Úmsgr r r Ú!test_add_series_with_period_index¢ s z6TestSeriesArithmetic.test_add_series_with_period_indexz%target_add,input_value,expected_valueú!ZhelloZworldzhello!zworld!rT ZhellomZworldmc C s( t |}|| }t |}t ||¡ d S r ©r r0 r[ )r> Z
|
target_addZ input_valueZexpected_valuerj rB rC r r r Útest_string_addition² s z)TestSeriesArithmetic.test_string_additionc C sº t dddtjgddddgd}t dtjdtjgddddgd}| |¡}t||}t |d |d ¡ t |d |d ¡ | |¡}t||}t |d |d ¡ t |d |d ¡ d S )
|
Nr8 rj rk ÚcÚdrq r. Úer )r r! rY Údivmodr0 r[ Zrdivmod)r> rj rk rB rC r r r Ú test_divmodÁ s
|
|
|
|
z TestSeriesArithmetic.test_divmodrc Né c C sD t tdd}t d|d}d}tjt|d || W 5 Q R X d S )Nr8 é
|
Úfoorq z4not all arguments converted during string formattingr )r rf ry r Ú TypeError)r> rc Ús1Ús2r r r r Útest_series_integer_modÐ s
|
z,TestSeriesArithmetic.test_series_integer_modc C sZ t ddgddgd}t ddgddgd}|| }t ddtjgdddgd}t ||¡ d S )Nr8 r. rq r é é )r r! rY r0 r[ )r> r r rB rC r r r Ú test_add_with_duplicate_indexÛ s
|
z2TestSeriesArithmetic.test_add_with_duplicate_indexc C sl t tdtdgtdddtdddgd}|| d¡ }| d¡| }t|d sXtt|d shtd S )Nz1.3z2.3iÜ r8 r. rq r )r r r Úshiftr rS )r> r\ rB Úresult2r r r Útest_add_na_handlingã s ÿz)TestSeriesArithmetic.test_add_na_handlingc C sN t g tg tjd}|| }t |¡ ¡ s.t|| ¡ }t|dksJtd S )N)rc Údtyper ) r r
|
r! Úfloat64Úisnanrb rS Úcopyr$ )r> Údatetime_seriesÚemptyrB r r r Útest_add_corner_casesí s
|
z*TestSeriesArithmetic.test_add_corner_casesc C sZ | t¡d d
}|| }t|jd d
|j |jd d
dd}t |d d
|¡ d S )Néûÿÿÿr( ©rc rQ )ÚastypeÚintr Úvaluesrc r0 r[ )r> r§ Zint_tsÚaddedrC r r r Útest_add_float_plus_intö s ýz,TestSeriesArithmetic.test_add_float_plus_intc C s> t g g tjd}t ddi}t || t tjgdgd¡ d S )N©r£ r* g rq )r r! Úint32r0 r[ rY )r> r r r r r Ú test_mul_empty_int_corner_case s z3TestSeriesArithmetic.test_mul_empty_int_corner_casec C s t tdddd}tj|jd< |d d d
}t tdtdtjg}|| }t ||¡ t |dd}|
|
¡ |
|
¡ d }t ||¡ d S ) Nz2012-1-1rN r ©Úperiodsr r. éÿÿÿÿr rP ) r r r! rY r r r ZNaTr0 r[ Úto_frame)r> ÚdtZdt2rC rB r r r Útest_sub_datetimelike_align s z0TestSeriesArithmetic.test_sub_datetimelike_alignc C sX t dddd}| d¡}td|d}td|d}|| |j|ksFt|j|ksTtd S )Nz
|
2016-01-01r ZCET)rµ ÚtzÚUTCrq )r Ú
|
tz_convertr rc rS )r> ÚdtiZdti_utcr\ Úser_utcr r r Ú test_alignment_doesnt_change_tz s
|
z4TestSeriesArithmetic.test_alignment_doesnt_change_tzc C s t ddddddg}td|d}td|d d
d}|| }dgd dgd dg }tj||jd }d
|
gd tjg }t||}t ||¡ d S ) NZ3z53ZLoJGZN503r. rq r¶ rV r )Ú
|
categoriesrr é ) r r r ZCategoricalIndexrÀ r! rY r0 r[ )r> ÚcatÚser1Úser2rB rl rm rC r r r Útest_alignment_categorical# s
|
z/TestSeriesArithmetic.test_alignment_categoricalc C sÀ dddddg}t tjdddd|d}t tjd dd|d}|| }t ddddddgd}t ||¡ t td
|
d d |d}t td d d |d}|| }t td dddddgd}t ||¡ d S )Nr. rN rV r8 rs Úint64r± rq r/ z20130101 09:00:00©rµ Z20130101z9 hours)r r! rv r0 r[ r r )r> rc r\ r@ rB rC r r r Ú$test_arithmetic_with_duplicate_index1 s z9TestSeriesArithmetic.test_arithmetic_with_duplicate_indexc C sL t dtjgdd}t ddgdd}|| }t dtjgdd}t ||¡ d S )Nr Úfloatr± r8 ÚInt64ÚFloat64)r r! rY r ÚNAr0 r[ ©r> rà rÄ rB rC r r r Ú'test_masked_and_non_masked_propagate_naC s
|
z<TestSeriesArithmetic.test_masked_and_non_masked_propagate_nac C sl t dtjddgdd}t ddtjdg}|| }t dtjtjdgdd}t ||¡ || }t ||¡ d S )Né r/ rV rÊ r± rp rË )r r rÌ r! rY r0 r[ rÍ r r r Ú+test_mask_div_propagate_na_for_non_na_dtypeK s z@TestSeriesArithmetic.test_mask_div_propagate_na_for_non_na_dtypez
|
val, dtype)rN rÊ )g @rË c C s` t dd dgdd}|dd |g }t dd d| g|d}t ||¡ dd |g| }t ||¡ d S )Nr8 rN rÊ r± r. r )r> Úvalr£ r\ rB rC r r r Ú test_add_list_to_masked_arrayV s z2TestSeriesArithmetic.test_add_list_to_masked_arrayc C s¢ |j jjdkrtrtnd }tdd dgdd}t |¡ |dd dg }W 5 Q R X tdd dgdd}t ||¡ t |¡ dd dg| }W 5 Q R X t ||¡ d S )Nr TFÚbooleanr± ) ÚnodeZcallspecÚidr Ú UserWarningr r0 Zassert_produces_warningr[ )r> r Úwarningr\ rB rC r r r Ú%test_add_list_to_masked_array_booleana s ÿÿý z:TestSeriesArithmetic.test_add_list_to_masked_array_boolean)rI rJ rK r ry rz r{ r r rf r r r¢ r© r° r³ r¹ r¿ rÅ rÈ rÎ rÐ rÒ rØ r r r r r} s2 þþ
|
|
|
|
|
|
r} c @ s e Zd Zej ddddg¡dd Zdd Zej d d
|
d
|
d d
|
gd fd d d
|
d gd fd
|
d
|
d d
|
gdfd
|
d
|
d
|
d
|
gdfd
|
d d d
|
gdfd
|
d d
|
d
|
gdfg¡dd Zej dd
|
d
|
d d gd dfd d d
|
d
|
gd dfd
|
d
|
d d gddfd
|
d
|
d
|
d gddfd d d d
|
gddfd d d
|
d
|
gddfg¡dd Z dS )ÚTestSeriesFlexComparisonru r Nrc c C sN t tj d¡}t tj d¡}t||j||d}|||}t ||¡ d S )Nr rt )r r! r" r r; rI r0 r[ )r> ru Ú comparison_opÚleftÚrightrB rC r r r Útest_comparison_flex_basicx s
|
|
z3TestSeriesFlexComparison.test_comparison_flex_basicc C sV t tj d¡}t tj d¡}d}tjt|d t||j|dd W 5 Q R X d S )Nr z No axis named 1 for object typer r8 rt ) r r! r" r ry r Ú
|
ValueErrorr; rI )r> rÚ rÛ rÜ r r r r Útest_comparison_bad_axis s
|
z1TestSeriesFlexComparison.test_comparison_bad_axisz
|
values, opFTÚeqÚneÚleÚltÚgeÚgtc C sZ t dddgtdd}t dddgtdd}t|||}t |tdd}t ||¡ d S )Nr8 rN r. Úabcrq ÚbcdÚabcd©r Úlistr; r0 r[ )r> r® rA rÛ rÜ rB rC r r r Ú test_comparison_flex_alignment s
|
z7TestSeriesFlexComparison.test_comparison_flex_alignmentzvalues, op, fill_valuer. c C s^ t dddgtdd}t dddgtdd}t||||d}t |tdd}t ||¡ d S ) Nr8 rN r. ræ rq rç rW rè ré )r> r® rA rX rÛ rÜ rB rC r r r Ú#test_comparison_flex_alignment_fill s
|
z<TestSeriesFlexComparison.test_comparison_flex_alignment_fill)
|
rI rJ rK ry rz r{ rÝ rß rë rì r r r r rÙ w s2
|
úþ
|
úþ rÙ c
|
@ s( e Zd Zdd Zej ddddddd g¡d
|
d Zej ddddddd g¡d d Zej ddddg¡dd Z dd Z
|
dd Z dd Z dd Z d d Zd d Zd d! Zej d"ed#d$d%ged&d'd(ed$d$d$ged)d'd(fed#d$d%ged&d'd(ed$d$d$d$ged*d'd(fg¡d+d, Zd-d. Zd/S )0ÚTestSeriesComparisonc C s t dddg}t ddg}d}tjt|d ||k W 5 Q R X t ddg}t ddd g}tjt|d ||k W 5 Q R X d S )
|
Nrj rk r z'only compare identically-labeled Seriesr r8 r. rN rV )r ry r rÞ )r> rj rk r r r r Ú test_comparison_different_length® s z5TestSeriesComparison.test_comparison_different_lengthr1 rà rá rå rã rä râ c C sD t dddgtdd}d}t|||j}t d¡}||ks@td S )Nr8 rN r. rq Úbool)r rf r; Údtypesr! r£ rS )r> r1 r\ ÚconstrB rC r r r Ú test_ser_flex_cmp_return_dtypesº s
|
|
z4TestSeriesComparison.test_ser_flex_cmp_return_dtypesc C sR t dddgtdd}|jd d
}d}t|||j}t d¡}||ksNtd S )Nr8 rN r. rq r rï )r rf r r; rð r! r£ rS )r> r1 r\ r¨ rñ rB rC r r r Ú%test_ser_flex_cmp_return_dtypes_emptyà s
|
z:TestSeriesComparison.test_ser_flex_cmp_return_dtypes_emptyÚnames)NNN)r ÚbarN)Úbazrö rö c
|
C sn |}t ddd|d d}t| |d ¡}|||}|j|d ksFt| d¡}tj|d d
|
}t| |d ¡}|||}|j|d kst|| d¡ }t| |d ¡}|||}|j|d ksÈt|t j
|
t j fkr tj dd|d d }t| |d ¡}|||}|j|d ks t|t j
|
t j fkrj| d ¡} t| |d ¡}||| }|j|d ksjtd S ) Nz1949-06-07 03:00:00ÚHr/ r )r rµ rQ r8 r. z
|
US/CentralZinferr )Ústartrµ rQ Úcategory)r r r: rQ rS Ú tz_localizer Z DatetimeIndexr r< rà rá Zinterval_ranger¬ )
|
r> rô rÚ rA r½ r\ rB ZtdiÚiiZcidxr r r Útest_ser_cmp_result_namesÍ s0
|
|
|
|
|
|
|
z.TestSeriesComparison.test_ser_cmp_result_namesc C sÀ t j d¡}t j d¡}t j|d d
< t ||¡}t jdd ||k d¡}W 5 Q R X t j|d d
< t ||¡ t
|
dddg}t
|
d d
|
d g}t
|
d d d g}t ||k|¡ t ||k|¡ d S ) Nr rN ra )ÚinvalidÚOrj rk r FT) r! r" r rY r Znangtrg r¬ r0 r= r r[ )r> rÛ rÜ rB rC Úsr r] r r r Útest_comparisonsõ s z%TestSeriesComparison.test_comparisonsc C sä t tddd}t tddd}t dddgdd}t tddd}ttd}ttd}|dk ¡ rjt|dk|dk k ¡ std|k ¡ rt|dkd s¤td|kd s´td|kd rÄt||k ¡ sÔt||k ¡ rät|t|k ¡ søt||k ¡ s
|
t||k ¡ s t||k ||kk ¡ s8t||k ||kk ¡ sTt||k ¡ rft||k ¡ rxt||k ¡ rt||k ¡ rt||k ¡ s®t||k ¡ sÀt||k ¡ rÒt||k ¡ rät||k ||kk ¡ s t||k ||kk ¡ s t||k ||kk ¡ s8t||k ||kk ¡ sTtd
|
}tjt|d ||k W 5 Q R X tjt|d ||k W 5 Q R X tjt|d ||k W 5 Q R X tjt|d ||k W 5 Q R X d S ) Nræ rù r± Úobjectrj rk ÚccZacbr ú can only compare equality or notr )r rê r rb rS ry r r )r> rj rk r r r Úfr r r r Ú test_categorical_comparisons sP z1TestSeriesComparison.test_categorical_comparisonsc C sD t ttd}d}tjt|d |dk W 5 Q R X t ttddd}tjt|d |dk W 5 Q R X t ttddd}d}tjt|d |d k W 5 Q R X tjt|d |d k W 5 Q R X tjt|d d |k W 5 Q R X tjt|d d |k W 5 Q R X t |d kt dddg¡ t |d kt dddg¡ d S )
|
Nræ r r rk F)ZorderedTz1Invalid comparison between dtype=category and strr )r r rê ry r r r0 r[ )r> r r r r r Ú5test_unequal_categorical_comparison_raises_type_errorD s& zJTestSeriesComparison.test_unequal_categorical_comparison_raises_type_errorc C sÜ t ddg}|dk}t ddg}t ||¡ |dk}t ddg}t ||¡ |dk}t ddg}t ||¡ |dk}t ddg}t ||¡ t ddg}|dk}t ddg}t ||¡ |dk}t ddg}t ||¡ d S )N)r8 r8 )r8 r. FT)r r r )r> rÿ rB rC r r r Útest_comparison_tuplesc s( z+TestSeriesComparison.test_comparison_tuplesc C sD t tdgtddgg}|tdgk}t ddg}t ||¡ d S )Nr8 r. TF)r Ú frozensetr0 r[ ©r> r\ rB rC r r r Útest_comparison_frozenset s z.TestSeriesComparison.test_comparison_frozensetc C s t tdddtd}tj|d d d
< |d }|||}|| ¡ | |j¡}|tj krh|
|
d¡ t ¡}n|
|
d¡ t ¡}t ||¡ d S ) Nr~ r rÇ r± r. r/ TF)r r r r! rY Zdropnare rc r< rá rZ r¬ rï r0 r[ )r> rÚ r\ rÑ rB rC r r r Ú"test_comparison_operators_with_nas s
|
|
z7TestSeriesComparison.test_comparison_operators_with_nasc C sb t dddddgdddddgtd}dddddg}t |jdk|¡sFtt |jdk |¡s^td S ) NrN rV r/ rs é r± TF)r rÉ r0 Z equalContentsrc rS )r> r( rC r r r Útest_ne s "z TestSeriesComparison.test_nez left, rightr8 r. rN ÚABCr* r« ZABDÚABCDc C s |t k r(d|j d}| ¡ }| ¡ }nd|j d}tjt|d ||k W 5 Q R X tjt|d ||k W 5 Q R X tjt|d ||k W 5 Q R X tjt|d ||k W 5 Q R X tjt|d ||k W 5 Q R X tjt|d ||k W 5 Q R X d S )Nz@Can only compare identically-labeled \(both index and columns\) z objectsz%Can only compare identically-labeled r )r rI r· ry r rÞ )r> rÛ rÜ Zframe_or_seriesr r r r Útest_comp_ops_df_compat s& ÿ
|
ÿz,TestSeriesComparison.test_comp_ops_df_compatc C s4 t dddg}|dk}t dddg}t ||¡ d S )NZ IntervalAZ IntervalBZ IntervalCTFr r r r r Ú$test_compare_series_interval_keywordÌ s z9TestSeriesComparison.test_compare_series_interval_keywordN)rI rJ rK rî ry rz r{ rò ró rü r r r r r
|
r r r rê r r r r r r rí s:
|
|
ÿ
|
%8 þþûþ
|
rí c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
ÚTestTimeSeriesArithmeticc C s¾ t ddddd}tj d¡d d
}ttj d¡| |¡ d¡d}tj d¡d d
}ttj d¡| |¡ d ¡d}|| }| d¡}| d¡}|| }|jj t
|
j ks®t t ||¡ d S )
|
Nú1/1/2011éd r÷ Úutc)rµ r rº éZ z
|
US/Easternrq z Europe/Berlin)r r! r" r# r r r r¼ rc rº r r rS r0 r[ ) r> r Úpermrà rÄ rB Zuts1Zuts2rC r r r Ú+test_series_add_tz_mismatch_converts_to_utcÛ s"
|
ÿ
|
ÿ
|
|
zDTestTimeSeriesArithmetic.test_series_add_tz_mismatch_converts_to_utcc C s| t dddd}ttj t|¡|d}| d¡}d}tjt |d || W 5 Q R X tjt |d || W 5 Q R X d S ) Nr r r÷ r´ rq r z0Cannot join tz-naive with tz-aware DatetimeIndexr )
|
r r r! r" r r$ rú ry r Ú Exception)r> r r\ r¾ r r r r Ú"test_series_add_aware_naive_raisesñ s
|
z;TestTimeSeriesArithmetic.test_series_add_aware_naive_raisesc C sJ t tddd}tjjdd}|| }t t dddg¡}t ||¡ d S ) Nz
|
2012-01-01rN rÇ rs )Údaysz
|
2011-12-26z
|
2011-12-27z
|
2011-12-28)r r r ÚoffsetsZ
|
DateOffsetÚ to_datetimer0 r[ )r> r? ÚoffsetrB rC r r r Útest_datetime_understoodþ s
|
z1TestTimeSeriesArithmetic.test_datetime_understoodc C s t ddd}ttj d¡|d}|dd
}| ¡ }dd |jD |_|| }|| }||dd
}|j d ¡|_t ||¡ t ||¡ d S )Nr~ é rÇ rq r/ c S s g | ] }| ¡ qS r )r )Ú.0r* r r r Ú
|
<listcomp> s zWTestTimeSeriesArithmetic.test_align_date_objects_with_datetimeindex.<locals>.<listcomp>)
|
r r r! r" r r¦ rc Z
|
_with_freqr0 r[ )r> r r( Zts_sliceZts2rB r¡ rC r r r Ú*test_align_date_objects_with_datetimeindex s zCTestTimeSeriesArithmetic.test_align_date_objects_with_datetimeindexN)rI rJ rK r r r r# r r r r r Ú s r c @ sT e Zd Zej deeej e
|
e e j g¡ej dddg¡dd Z dd Zd d
|
Zd S ) ÚTestNamePreservationÚboxÚflexTFc C sÎ |}t td|d d}t td|d d}|j d¡}|dk} ||}|rd| rTd S t|||}
|
n
|
|||}
|
t|
|
t s|t|tt fkrª|
|
j|d ksÊ|
|
j|d ksÊtn |
|
j|d ksÊ|
|
j|d ksÊtd S )Nr r rP r8 Ú_)ÚandZrandÚxorZrxorÚorZrorr. ) r rf rI Ústripr; Ú
|
isinstancerS r
|
rQ ) r> r& r% rô Zall_binary_operatorsrA rÛ rÜ rQ Z
|
is_logicalrB r r r Ú test_series_ops_name_retention s
|
"z3TestNamePreservation.test_series_ops_name_retentionc C s || }|j |j kst| |¡}|j |j ks2t||d d
}|j |j ksRt| ¡ }d|_ || }|j d ksvt| |¡}|j d kstddddddd d
|
g}|d d |D }|D ]T}| ¡ }t|||}|j |j ksât| ¡ }d |_ t|||}|j d ks¸tq¸d S )Néþÿÿÿzsomething elser2 r3 r4 r| r6 r5 Úmodr7 c S s g | ] }d | qS )r9 r )r! rA r r r r" G s zGTestNamePreservation.test_binop_maybe_preserve_name.<locals>.<listcomp>Úchanged)rQ rS r4 r¦ r2 r; )r> r§ rB Úcpr rA r\ r r r Ú test_binop_maybe_preserve_name4 s,
|
|
z3TestNamePreservation.test_binop_maybe_preserve_namec C s |d }|j |j kstd S r- )rQ rS )r> r§ rB r r r Útest_scalarop_preserve_nameT s z0TestNamePreservation.test_scalarop_preserve_nameN)rI rJ rK ry rz r{ rê Útupler! Úarrayr
|
r r r- r2 r3 r r r r r$ s
|
r$ c @ s" e Zd Zej dd¡dd ZdS )ÚTestInplaceOperationsz)dtype1, dtype2, dtype_expected, dtype_mul))rÊ rÊ rÊ rÊ )rÉ rÉ rÉ rÉ )rÊ rÉ rË rË )rÊ rË rË rË c C s t dg|d}t dg|d}||7 }t dg|d}t ||¡ ||8 }t dg|d}t ||¡ ||9 }t dg|d}t ||¡ d S )Nr8 r± r. rN r )r> Zdtype1Zdtype2Zdtype_expectedZ dtype_mulrà rÄ rC r r r Útest_series_inplace_opsZ s z-TestInplaceOperations.test_series_inplace_opsN)rI rJ rK ry rz r{ r7 r r r r r6 Y s
|
þ r6 c C sv |}t |dk r&| j tjjdd¡ tj|jd< |d k}|j d rHt
|
|j d rVt
|
|d k}|j d slt
|
|j d szt
|
d |k}|j d rt
|
|j d rt
|
d |k}|j d s´t
|
|j d sÂt
|
t |j sØt |j r"d}tjt|d d |k W 5 Q R X tjt|d |d k W 5 Q R X nPd |k}|j d r:t
|
|j d rJt
|
|d k }|j d rbt
|
|j d rrt
|
d S )Nr8 z%Test doesn't make sense on empty data)Úreasonr zInvalid comparisonr )r$ rÔ Z
|
add_markerry rz Zxfailr! rY r ZiatrS r r£ r r r )r Zseries_with_simple_indexr? rB r r r r Útest_none_comparisonu s< ÿ r9 c C sÎ t tdtjjtdtdddggdddgdd} t d d
|
tdd D tjjtdddggddgdd}| jtjd gd d
d d
f | }t d dddgtjjddddgdddgdd}t
|
||¡ d S )Né ÚabZxyr8 r. Únum)rô rq c S s g | ] }d | qS )éè r )r! rn r r r r" ® s z;test_series_varied_multiindex_alignment.<locals>.<listcomp>r/ rj r= iÑ iº i£ )r* r8 rj )r* r. rj )r` r8 rj )r` r. rj ) r rf r Z
|
MultiIndexZ from_productrê ÚlocZ
|
IndexSliceÚ from_tuplesr0 r[ )r r rB rC r r r Ú'test_series_varied_multiindex_alignment¥ s( ÿþ þ$
|
|
þþr@ c C s2 t dgd d¡} t dgd }t | |¡ d S )Nr. i' r¶ r8 )r Zrmodr0 r[ )rB rC r r r Ú!test_rmod_consistent_large_series¼ s rA )0Údatetimer r r Údecimalr r< Únumpyr! ry Zpandas._libs.tslibsr Zpandas.core.dtypes.commonr r Zpandasr r r
|
r r r r r Zpandas._testingZ_testingr0 Z pandas.corer r Zpandas.core.computationr r Z pandas.core.computation.checkr Zfixturer r& r' r} rÙ rí r r$ r6 r9 r@ rA r r r r Ú<module> s8 $
|
m Y6 /=B 0
|
