U
|
¬ý°d ã @ sx d Z ddlZddlZddlmZmZmZmZ ddl m
|
Z ddl m Z mZ dddZddd Zded d d Zdd ZdS )z)
|
Assertion helpers for arithmetic tests.
|
é N)Ú DataFrameÚIndexÚSeriesÚarray)Ú BooleanArrayÚ PandasArrayú
|
cannot addc C sH t jt|d | | W 5 Q R X t jt|d || W 5 Q R X dS )z§
|
Helper to assert that left and right cannot be added.
|
|
Parameters
|
----------
|
left : object
|
right : object
|
msg : str, default "cannot add"
|
©ÚmatchN©ÚpytestÚraisesÚ TypeError©ÚleftÚrightÚmsg© r úUd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arithmetic/common.pyÚassert_cannot_add s
|
r c C s t jt|d | | W 5 Q R X t jt|d || W 5 Q R X t jt|d | | W 5 Q R X t jt|d || W 5 Q R X dS )z»
|
Helper to assert that left and right can be neither added nor subtracted.
|
|
Parameters
|
----------
|
left : object
|
right : object
|
msg : str or None, default None
|
r Nr r r r r Úassert_invalid_addsub_type$ s
|
r F)Úis_cmpc C sl t | tst |trtS t | ts,t |trD|r@t | tr@tjS tS t | tsXt |trf|rbtjS tS tjS )zÛ
|
Get the box to use for 'expected' in an arithmetic or comparison operation.
|
|
Parameters
|
left : Any
|
right : Any
|
is_cmp : bool, default False
|
Whether the operation is a comparison method.
|
)Ú
|
isinstancer r r Únpr ÚtmZto_array)r r r r r r Úget_upcast_box8 s r c C sæ |t tfkr|ntj}dd }|}t|t r<t| tr<tj}|| |k}|tj|jtjd}t ||¡ ||| k}t |||¡ || |k}t || ¡ ||| k}t ||| ¡ d
|
ddddd g¡}t j t |d
|
| |k W 5 Q R X t j t |d
|
| |k W 5 Q R X t j t |d
|
| |k W 5 Q R X t j t |d
|
| |k W 5 Q R X t j t |d
|
|| k W 5 Q R X t j t |d
|
|| k W 5 Q R X t j t |d
|
|| k W 5 Q R X t j t |d
|
|| k W 5 Q R X d S ) a
|
Assert that comparison operations with mismatched types behave correctly.
|
|
Parameters
|
----------
|
left : np.ndarray, ExtensionArray, Index, or Series
|
right : object
|
box : {pd.DataFrame, pd.Series, pd.Index, pd.array, tm.to_array}
|
c S s( t | tr| jS t | tr$| t¡S | S )N)r r Z_ndarrayr ZastypeÚbool)Úxr r r Úxbox2` s
|
|
|
|
z(assert_invalid_comparison.<locals>.xbox2)Zdtypeú|zInvalid comparison betweenzCannot compare typeznot supported betweenzinvalid type promotionz¿The DTypes <class 'numpy.dtype\[datetime64\]'> and <class 'numpy.dtype\[int64\]'> do not have a common DType. For example they cannot be stored in a single array unless the dtype is `object`.r N)r r r r r ÚzerosÚshapeZbool_r Z assert_equalÚjoinr r r ) r r ZboxZxboxr Zrev_boxÚresultÚexpectedr r r r Úassert_invalid_comparisonQ sL ùÿr% )r )N)F)Ú__doc__Únumpyr r Zpandasr r r r Zpandas._testingZ_testingr Zpandas.core.arraysr r r r r r r% r r r r Ú<module> s
|
|
|
