U
|
®ý°dr= ã @ s d dl mZ d dlZd dlZd dlZd dlZd dlm m Z
|
d dl Z d dl mZ d dlmZ ejejgZddgZddgZejdd Zej d
|
ejej ej g¡ejjdeed d d Z ej d
|
e¡ejjdeed ejjdddgddgd dd Z ej d
|
e¡ejjdeed ejjdddgddgd dd Z ej d
|
e¡ejjdeed ejjdddgddgd ejjdddgddgd dd Z!ej d
|
e¡ejjdeed ej dddg¡dd Z"ej d
|
ej#g¡ejjdeed ej dddg¡ej $d ¡d d Z%ejjdeed d d Z&ejjdeed ej d
|
e¡d!d" Z'd#d$ Z(eje j)d%d&d'gej*d(e j)d%d&d'gd)d(e j)d%d&d'gd*d(e j)d%d+d'gd,d(e +d-d.d/g¡e +d-d.d/g¡ ,d0¡e +d-d.d/g¡j-d1d2e .d3d4d5g¡e /e 0d d%¡e 0d'd&¡e 0d%d'¡g¡g d6d7 d8d9d: Z1G d;d< d<Z2ej d=e3ee4g¡d>d? Z5d@dA Z6dBdC Z7dDdE Z8dFdG Z9e
|
j:dHdI Z;dS )Jé )ÚdequeN)Ú SparseArrayTFÚsparseZdensec C sL t jjddddd} t jjddddd}d| ddd
< d|ddd
< | |fS ) zO
|
A pair of random, length-100 integer-dtype arrays, that are mostly 0.
|
r é
|
éd Úint64©ÚdtypeNé é )ÚnpÚrandomÚrandint)Úa1Úa2© r úUd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/test_ufunc.pyÚarrays_for_binary_ufunc s
|
r Úufunc)Úidsc C s t jjddddd}d|d d d
< |r:t|t dd¡d}ttjd d
}d}tj |||d}| |}tj | |||d}t
|
||¡ d S )Nr r r r é Úname©Úindexr ) r r r r ÚpdÚ SparseDtypeÚlistÚstringÚ ascii_lettersÚSeriesÚtmÚassert_series_equal)r r Úarrr r ÚseriesÚresultÚexpectedr r r Útest_unary_ufunc s r& ÚflipZflippedZstraightc C s |\}}|r4t |t dd¡d}t |t dd¡d}d}tj||d}|}||f} ||f}
|
| rnt| } t|
|
}
|
tj|| |d} ||
|
} t | | ¡ d S ©Nr r r r ©r )r r r r Úreversedr r! © r' r r r r r r r# ÚotherÚ
|
array_argsÚ series_argsr% r$ r r r Ú test_binary_ufunc_with_array1 s r/ c C s¨ |\}}|r4t |t dd¡d}t |t dd¡d}d}tj||d}tj||d d¡}||f} ||f}
|
| r~t| } t|
|
}
|
tj|| |d} ||
|
} t | | ¡ d S r( ) r r r r ÚIndexZastyper* r r! r+ r r r Ú test_binary_ufunc_with_indexK s r1 ÚshuffleZ unalignedZalignedc C sø |\}}|r4t |t dd¡d}t |t dd¡d}d}tj||d}tj||d} tj t|¡}
|
|r| |
|
¡} | r| |¡d j
|
} q | | ¡d j
|
} n|j
|
} ||f} || f} | rÌt t | } t t | } tj|| | |d}|| }t ||¡ d S )Nr r r r r) r )r r r r r r Z permutationÚlenZtakeZalignr Útupler* r r! )r' r2 r r r r r r r# r, Úidxr r- r. r% r$ r r r Ú test_binary_ufunc_with_seriesh s,
|
|
r6 c C s| |\}}|rt |}d}tj|dd}||f}||f} |rRtt|}tt| } tj| | dd}
|
| | } t | |
|
¡ d S )Nr r r) )r r r r4 r* r r! ) r r r' r r" Ú_r, r# r. r- r% r$ r r r Útest_binary_ufunc_scalar s r8 z$ignore:divide by zero:RuntimeWarningc
|
C sØ |\}}d||dk< d||dk< |rLt |t dd¡d}t |t dd¡d}t |¡}t |¡}|rp|jdd}| ||}t|tst| ||} t| ts tt | d t |d ¡¡ t | d t |d ¡¡ d S )Né r r r )Úfrac)
|
r r r r ÚsampleÚ
|
isinstancer4 ÚAssertionErrorr r! )
|
r r r2 r r r Ús1Ús2r% r$ r r r Ú"test_multiple_output_binary_ufuncs¬ s
|
|
|
|
r@ c C s |\}}| rt |}tj|dd}t |¡}t |¡}t|tsDtt|tsRtt |d tj|d dd¡ t |d tj|d dd¡ d S )Nr r) r r9 )
|
r r r r Úmodfr< r4 r= r r! )r r r" r7 r# r$ r% r r r Útest_multiple_output_ufuncÎ s
|
|
rB c C s@ |\}}t j|dd}t j|dd}| ||}|jd ks<td S )NÚar) Úb)r r r r= )r r r r r r> r? r$ r r r Ú#test_binary_ufunc_drops_series_nameâ s
|
|
rE c
|
C s~ G dd d} t | d| dg¡}t |¡}t t ||¡t t ||¡¡¡ t t || d¡t t || d¡¡¡ d S )Nc @ s" e Zd ZddddZdd ZdS )z$test_object_series_ok.<locals>.DummyN©Úreturnc S s
|
|| _ d S ©N©Úvalue©ÚselfrJ r r r Ú__init__ð s z-test_object_series_ok.<locals>.Dummy.__init__c S s | j |j S rH rI ©rL r, r r r Ú__add__ó s z,test_object_series_ok.<locals>.Dummy.__add__)Ú__name__Ú
|
__module__Ú __qualname__rM rO r r r r ÚDummyï s rS r r9 )r Úarrayr r r r! Úadd)rS r" Úserr r r Útest_object_series_okî s
|
|
"rW r9 r
|
r r ZInt64ZFloat32r z Sparse[int]Ú2000Z2010Z2001ZCETÚD)Úfreqz1 Dayz3 Daysz2 Daysc C s
|
t | jS rH )Ústrr )Úxr r r Ú<lambda> ó r] )Úparamsr c C s | j S rH )Úparam)Úrequestr r r Úvalues_for_np_reduceü s rb c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
ÚTestNumpyReductionsc C sþ |}|}t d ¡ ||}W 5 Q R X t|tjjjrPtjj dd}|j
|
|¡ |j j dkrÄtj |¡}|tjkr|jdd} t || ¡ qú|tjkr®|j ¡ } || ksÂtqú| ¡ } || ksútn6d ddd g¡}
|
tjt|
|
d
|
tj |¡ W 5 Q R X d S ) NzSparseArray has no 'prod')ÚreasonZiufF©Z numeric_onlyú|údoes not support reductionúunsupported operand typez$ufunc 'multiply' cannot use operands©Úmatch)r Úassert_produces_warningr< r ÚcoreZarraysr ÚpytestÚmarkZxfailÚnodeZ
|
add_markerr Úkindr ÚmultiplyÚreduceÚ DataFrameÚprodr! r0 Ú_valuesr= ÚjoinÚraisesÚ TypeError) rL rb Úbox_with_arrayra ÚboxÚvaluesÚobjrn r$ r% Úmsgr r r Ú test_multiply s2
|
|
|
ýÿz!TestNumpyReductions.test_multiplyc C sÔ |}|}t d ¡ ||}W 5 Q R X |jjdkrtj |¡}|tjkrb|j dd}t
|
||¡ qÐ|tj kr|j ¡ }||kst qÐ| ¡ }||ksÐt n6d dddg¡}tjt|d tj |¡ W 5 Q R X d S ) NZmiufFre rf rg rh z ufunc 'add' cannot use operandsri )r rk r rp r rU rr r rs Úsumr! r0 ru r= rv rm rw rx ) rL rb ry rz r{ r| r$ r% r} r r r Útest_add4 s,
|
|
|
ýÿz TestNumpyReductions.test_addc C s¤ |}|}d}|t jkr&|jjdkr&d}t d ¡ ||}W 5 Q R X tj |¡}|t j krt|j
|
dd}t ||¡ n,|d }||kst |r t |t |ks t d S )NT©ÚiÚfFre r9 )r r0 r rp r rk r Úmaximumrr rs Úmaxr! r= Útype© rL rb ry rz r{ Z same_typer| r$ r% r r r Útest_maxR s
|
z TestNumpyReductions.test_maxc C s¤ |}|}d}|t jkr&|jjdkr&d}t d ¡ ||}W 5 Q R X tj |¡}|t j krt|j
|
dd}t ||¡ n,|d }||kst |r t |t |ks t d S )NTr Fre r )r r0 r rp r rk r Zminimumrr rs Úminr! r= r r r r r Útest_minj s
|
z TestNumpyReductions.test_minN)rP rQ rR r~ r r r r r r r rc s " rc Útype_c C sX t jdddgdd}| dddg}t ||¡}t jt | ¡ |¡dd}t ||¡ d S )Nr9 r r
|
r r) r é )r r r rU Zto_numpyr r! )r rC rD r$ r% r r r Ú test_binary_ufunc_other_types s
|
r c s^ G fddd t d dg¡} t | d¡}t d dg¡}t ||¡ d S )Nc sB e Zd ZddddZdd Zed fdd Zedd d
|
ZdS ) z#test_object_dtype_ok.<locals>.ThingNrF c S s
|
|| _ d S rH rI rK r r r rM s z,test_object_dtype_ok.<locals>.Thing.__init__c S s t |d|}t| | j| S )NrJ )Úgetattrr rJ rN r r r rO s z+test_object_dtype_ok.<locals>.Thing.__add__c s t | ko| j|jkS rH )r rJ rN ©ÚThingr r Ú__eq__ s z*test_object_dtype_ok.<locals>.Thing.__eq__c S s d| j dS )NzThing(ú)rI )rL r r r Ú__repr__ s z,test_object_dtype_ok.<locals>.Thing.__repr__) rP rQ rR rM rO Úboolr r[ r r r r r r s r r9 r r
|
)r r r rU r r! )Úsr$ r% r r r Útest_object_dtype_ok s
|
r c C sN t dddg¡} t dddg¡}tjttjd tj
|
| |¡ W 5 Q R X d S )Nr9 r r
|
ri ) r r r rT rm rw ÚNotImplementedErrorr ZEMPTY_STRING_PATTERNÚsubtractÚouter)rV r| r r r Ú
|
test_outer¡ s r c C sP t jdddggd} t jdddgd}t jdgd}t | |¡}t ||¡ d S )Néÿÿÿÿr9 r )Údataéf )r rs r Úmatmulr Zassert_frame_equal)Zdf1Zdf2r% r$ r r r Útest_np_matmulª s
|
r c C s dd } t | dd¡}t ddg¡}|||d}tjddgtd}t ||¡ t ddgg¡}d}tj t
|
t |¡d |||| W 5 Q R X d S )
|
Nc S s | | | S rH r )r\ ÚyÚzr r r Úadd3¶ s z2test_array_ufuncs_for_many_arguments.<locals>.add3r
|
r9 r r r zTCannot apply ufunc <ufunc 'add3 (vectorized)'> to mixed DataFrame and Series inputs.ri ) r Z
|
frompyfuncr r Úobjectr r! rs rm rw r ÚreÚescape)r¢ r rV r$ r% Zdfr} r r r Ú$test_array_ufuncs_for_many_arguments´ s ÿr¦ c C s> t ddddg¡} t | ¡}t ddddg¡}t ||¡ d S ) Ng ø¿g à¿g à?g ø?g ð¿g g g ð?)r r r Zfixr r! )rV r$ r% r r r Ú test_np_fixË s
|
r§ )<Ú collectionsr r¤ r Únumpyr rm Z pandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandasr Zpandas._testingZ_testingr Z pandas.arraysr rU Z logaddexpZ BINARY_UFUNCSZSPARSEZ
|
SPARSE_IDSZfixturer rn Z parametrizeZpositiveÚfloorÚexpr& r/ r1 r6 r8 ÚdivmodÚfilterwarningsr@ rB rE rW rT r Ú to_datetimeZ tz_localizeZ to_periodZ to_timedeltaZ IntervalIndexZIntervalrb rc r r4 r r r r r¦ Z&skip_copy_on_write_not_yet_implementedr§ r r r r Ú<module> s
|
&
|
|
|
&÷ ô
|
s
|
|
|
