iMpdZddlZddlZddlmZmZmZmZddl m Z ddl m Z mZd dZd dZd defdZd Zy) z) Assertion helpers for arithmetic tests. N) DataFrameIndexSeriesarray) BooleanArrayNumpyExtensionArrayctjt|5||zdddtjt|5||zdddy#1swY3xYw#1swYyxYw)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 s/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/pandas/tests/arithmetic/common.pyassert_cannot_addrsK y , u  - y ,   - , - , , ,sAA A A)ctjt|5||zdddtjt|5||zdddtjt|5||z dddtjt|5||z dddy#1swYxYw#1swYhxYw#1swYKxYw#1swYyxYw)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 ,   - , - , , , , , , ,s/B&B2.B>C &B/2B;>C Cis_cmpcnt|tst|trtSt|tst|tr(|r t|trtj StSt|tst|tr|rtj StSt 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@sx$ "j &B$:eV#< ju-88O $*UE": 88O ;;cR|ttfvr|ntj}d}|||k(}|tj|jtj }t j|||||k(}t j||||||k7}t j|||||k7}t j|||djgd}tjt|5||kdddtjt|5||kdddtjt|5||kDdddtjt|5||k\dddtjt|5||kdddtjt|5||kdddtjt|5||kDdddtjt|5||k\dddy#1swY*xYw#1swYxYw#1swYxYw#1swYxYw#1swYxYw#1swYxYw#1swY}xYw#1swYyxYw)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|j t S|SN)rr_ndarrayrastypebool)xs rxbox2z(assert_invalid_comparison..xbox2hs6 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_equaljoinr rr)rrboxxboxr'resultexpectedrs rassert_invalid_comparisonr3YseU^+3D45= !FBHHV\\:;HOOFH% 5D= !FOOFDN+ 45= !FOOFXI& 5D= !FOOFD(O, ((  C y , u  - y ,   - y , u  - y ,   - y ,   - y ,   - y ,   - y ,   - , - , , , , , , , , , , , , , , ,s`II+I!I-=I9&JJ8JII!I*-I69JJJJ&)z cannot addr")F)__doc__numpyrr pandasrrrrpandas._testing_testingrpandas.core.arraysrrrrr%rr3rrr;sB   &02Er