ttjMrdZddlZddlZddlmZmZmZmZddl m Z ddl m Z mZd dZd dZdd efd Zd ZdS)z) Assertion helpers for arithmetic tests. N) DataFrameIndexSeriesarray) BooleanArrayNumpyExtensionArray cannot addctjt|5||zdddn #1swxYwYtjt|5||zddddS#1swxYwYdS)a Helper function to assert that two objects cannot be added. Parameters ---------- left : object The first operand. right : object The second operand. msg : str, default "cannot add" The error message expected in the TypeError. matchNpytestraises TypeErrorleftrightmsgs \/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/pandas/tests/arithmetic/common.pyassert_cannot_addrs y , , , u  y , , ,  s.22A''A+.A+ctjt|5||zdddn #1swxYwYtjt|5||zdddn #1swxYwYtjt|5||z dddn #1swxYwYtjt|5||z ddddS#1swxYwYdS)a1 Helper function to assert that two objects can neither be added nor subtracted. Parameters ---------- left : object The first operand. right : object The second operand. msg : str or None, default None The error message expected in the TypeError. r Nr rs rassert_invalid_addsub_typer(s y , , , u  y , , ,   y , , , u  y , , ,  sB.22A&&A*-A* BB"%B"CCCFis_cmpct|tst|trtSt|tst|tr*|r!t|tr tjStSt|tst|tr|r tjStSt jS)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)rrrs rget_upcast_boxr@s$ ""j &B&B$:eV#<#<  ju-- 8O $*UE":":  8O ;c|ttfvr|n tj}d}|||k}|tj|jtj}t j|||||k}t j||||||k}t j|||||k}t j|||dgd}tj t|5||kdddn #1swxYwYtj t|5||kdddn #1swxYwYtj t|5||kdddn #1swxYwYtj t|5||kdddn #1swxYwYtj t|5||kdddn #1swxYwYtj t|5||kdddn #1swxYwYtj t|5||kdddn #1swxYwYtj t|5||kddddS#1swxYwYdS)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} ct|tr|jSt|tr|t S|SN)rr_ndarrayrastypebool)xs rxbox2z(assert_invalid_comparison..xbox2hsF a, - - :  a & & "88D>> !r )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) rrrzerosshapebool_r assert_equaljoinrrr)rrboxxboxr(resultexpectedrs rassert_invalid_comparisonr4YseU^++33DU45= ! !FtBHV\:::;;HOFH%%% U5D= ! !FOFDDNN+++ U45= ! !FOFXI&&& U5D= ! !FOFDD(OO,,, ((   C y , , , u  y , , ,   y , , , u  y , , ,   y , , ,   y , , ,   y , , ,   y , , ,  s&D99D=D=E22E69E6F++F/2F/G$$G(+G( HH!$H!III<JJJ5K  K K )r r#)F)__doc__numpyrrpandasrrrrpandas._testing_testingrpandas.core.arraysrrrrr&rr4r rr<s   &02EEEEEr