a nEb@sxdZddlZddlZddlmZmZmZmZddl m Z ddl m Z mZdddZddd Zded d d ZddZdS)z) Assertion helpers for arithmetic tests. N) DataFrameIndexSeriesarray) BooleanArray PandasArray cannot addcCsptjt|d||Wdn1s,0Ytjt|d||Wdn1sb0YdS)z Helper to assert that left and right cannot be added. Parameters ---------- left : object right : object msg : str, default "cannot add" matchNpytestraises TypeErrorleftrightmsgr^/home/ja/django-apps/lartica_env/lib/python3.9/site-packages/pandas/tests/arithmetic/common.pyassert_cannot_adds &rcCstjt|d||Wdn1s,0Ytjt|d||Wdn1sb0Ytjt|d||Wdn1s0Ytjt|d||Wdn1s0YdS)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 &&&rF)is_cmpcCslt|tst|trtSt|ts,t|trD|r@t|tr@tjStSt|tsXt|trf|rbtjStStjS)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. ) isinstancerrrnprtmZto_array)rrrrrrget_upcast_box8s rc Cs|ttfvr|ntj}dd}|}t|tr.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`.r N)rrrrrZzerosshapeZbool_rZ assert_equaljoinr r r) rrboxZxboxrZrev_boxresultexpectedrrrrassert_invalid_comparisonQsB       &((((((r%)r)N)F)__doc__numpyrr ZpandasrrrrZpandas._testingZ_testingrZpandas.core.arraysrrrrrrr%rrrrs