o ۭj @sxdZddlZddlZddlmZmZmZmZddl m Z ddl m Z mZdddZddd Zdd efd d ZddZdS)z) Assertion helpers for arithmetic tests. N) DataFrameIndexSeriesarray) BooleanArrayNumpyExtensionArray cannot addcCsrtjt|d ||Wdn1swYtjt|d ||WddS1s2wYdS)z Helper to assert that left and right cannot be added. Parameters ---------- left : object right : object msg : str, default "cannot add" matchNpytestraises TypeErrorleftrightmsgr[/var/www/html/CropPilot/venv/lib/python3.10/site-packages/pandas/tests/arithmetic/common.pyassert_cannot_adds   "rcCstjt|d ||Wdn1swYtjt|d ||Wdn1s1wYtjt|d ||Wdn1sLwYtjt|d ||WddS1shwYdS)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 rrrrassert_invalid_addsub_type$s    "rFis_cmpcCslt|ts t|tr tSt|tst|tr"|r t|tr tjStSt|ts,t|tr3|r1tjStStjS)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. ) isinstancerrrnprtmto_array)rrrrrrget_upcast_box8s rc Cs|ttfvr|ntj}dd}|}t|trt|trtj}|||k}|tj|jtjd}t |||||k}t ||||||k}t |||||k}t |||d gd}t j t |d ||kWdn1s|wYt j t |d ||kWdn1swYt j t |d ||kWdn1swYt j t |d ||kWdn1swYt j t |d ||kWdn1swYt j t |d ||kWdn 1swYt j t |d ||kWdn 1s wYt j t |d ||kWddS1s=wYdS)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} cSs(t|tr|jSt|tr|tS|SN)rr_ndarrayrastypebool)xrrrxbox2`s   z(assert_invalid_comparison..xbox2)dtype|)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`.r N)rrrrrzerosshapebool_r assert_equaljoinr r r) rrboxxboxr"rev_boxresultexpectedrrrrassert_invalid_comparisonQsR               $r/)rr)F)__doc__numpyrr pandasrrrrpandas._testing_testingrpandas.core.arraysrrrrr rr/rrrrs