U f/e @sbdZddlZddlZddlmZmZmZmZddl m Z ddl m Z d ddZddZd d ZdS) z) Assertion helpers for arithmetic tests. N) DataFrameIndexSeriesarray) PandasArrayc Cstjt|d||W5QRXtjt|d||W5QRXtjt|d||W5QRXtjt|d||W5QRXdS)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)leftrightmsgrB/tmp/pip-unpacked-wheel-tiezk1ph/pandas/tests/arithmetic/common.pyassert_invalid_addsub_types rcCsF|tkst|trtS|tks(t|tr,tS|tks>t|trBtS|S)z? Given two box-types, find the one that takes priority )r isinstancerr)boxZvectorrrrget_upcast_box%src Cs|ttfkr|ntj}dd}|||k}|tj|jtjd}t|||||k}t|||||k}t|||||k}t||dddddd g}t j t |d ||kW5QRXt j t |d ||kW5QRXt j t |d ||kW5QRXt j t |d ||kW5QRXt j t |d ||kW5QRXt j t |d ||kW5QRXt j t |d ||kW5QRXt j t |d ||kW5QRXd 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} cSst|tr|jS|S)N)rrZ_ndarray)xrrrxbox2As z(assert_invalid_comparison..xbox2)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) rrnpzerosshapeZbool_tmZ assert_equaljoinr r r )r r rZxboxrresultexpectedrrrrassert_invalid_comparison2sF       r)N)__doc__Znumpyrr ZpandasrrrrZpandas._testingZ_testingrZpandas.core.arraysrrrrrrrrs