3 pdc @sPdZddlZddlZddlmZmZmZddlj Z d ddZ ddZ dd Z dS) z) Assertion helpers for arithmetic tests. N) DataFrameIndexSeriescCstjt|d||WdQRXtjt|d||WdQRXtjt|d||WdQRXtjt|d||WdQRXdS)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 )matchN)pytestraises TypeError)leftrightmsgr @/tmp/pip-build-7vycvbft/pandas/pandas/tests/arithmetic/common.pyassert_invalid_addsub_type s rcCsF|tkst|trtS|tks(t|tr,tS|tks>t|trBtS|S)z? Given two box-types, find the one that takes priority )r isinstancerr)boxZvectorr r r get_upcast_boxsrc:Cs|tk r |ntj}||k}|tj|jtjd}tj||||k}tj||||k}tj||||k}tj||djdddddg}t j t |d||kWd QRXt j t |d||kWd QRXt j t |d||kWd QRXt j t |d||kWd QRXt j t |d||kWd QRXt j t |d||kWd QRXt j t |d||kWd QRXt j t |d||kWd QRXd S) z 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, tm.to_array} )Zdtype|zInvalid comparison betweenzCannot compare typeznot supported betweenzinvalid type promotionzThe DTypes and do not have a common DType. For example they cannot be stored in a single array unless the dtype is `object`.)rN) rnparrayzerosshapeZbool_tmZ assert_equaljoinrrr)r r rZxboxresultexpectedr r r r assert_invalid_comparison,s@   r)N)__doc__numpyrrZpandasrrrZpandas._testingZ_testingrrrrr r r r s