j| ddlmZddlZddlmZddlmZddlmZddl m Z ddl m Z ej e5ddlmZdddn #1swxYwYd%dZedddedddedddddddddddd&d#Zedddedddedddddddddddd&d$ZdS)') annotationsN)cast)deprecate_renamed_parameter) DataFrame) LazyFrame)raise_assertion_error)assert_dataframe_equal_pyleftDataFrame | LazyFramerightreturnboolc&d}t|trt|trdSt|trt|trdStddt |jt |jdS)NTFinputszunexpected input types) isinstancerrrtype__name__)r r __tracebackhide__s a/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/polars/testing/asserts/frame.py_assert_correct_input_typers$ ""  z%'C'C  u D) $ $ E9)E)E t  $ JJ  KK      check_dtype check_dtypesz0.20.31)versionrtolrel_tolz1.32.3atolabs_tolTFgh㈵>g:0yE>check_row_ordercheck_column_orderr check_exactrrcategorical_as_strr r!r"floatr#Nonec d} t||} | r(||}}td|td|}}t|j|j||||||| dS)a Assert that the left and right frame are equal. Raises a detailed `AssertionError` if the frames differ. This function is intended for use in unit tests. .. versionchanged:: 0.20.31 The `check_dtype` parameter was renamed `check_dtypes`. .. versionchanged:: 1.32.3 The `rtol` and `atol` parameters were renamed to `rel_tol` and `abs_tol`, respectively. Parameters ---------- left The first DataFrame or LazyFrame to compare. right The second DataFrame or LazyFrame to compare. check_row_order Requires row order to match. check_column_order Requires column order to match. check_dtypes Requires data types to match. check_exact Requires float values to match exactly. If set to `False`, values are considered equal when within tolerance of each other (see `rel_tol` and `abs_tol`). Only affects columns with a Float data type. rel_tol Relative tolerance for inexact checking. Fraction of values in `right`. abs_tol Absolute tolerance for inexact checking. categorical_as_str Cast categorical columns to string before comparing. Enabling this helps compare columns that do not share the same string cache. See Also -------- assert_series_equal assert_frame_not_equal Notes ----- When using pytest, it may be worthwhile to shorten Python traceback printing by passing `--tb=short`. The default mode tends to be unhelpfully verbose. More information in the `pytest docs `_. Examples -------- >>> from polars.testing import assert_frame_equal >>> df1 = pl.DataFrame({"a": [1, 2, 3]}) >>> df2 = pl.DataFrame({"a": [1, 5, 3]}) >>> assert_frame_equal(df1, df2) Traceback (most recent call last): ... AssertionError: DataFrames are different (value mismatch for column "a") [left]: shape: (3,) Series: 'a' [i64] [ 1 2 3 ] [right]: shape: (3,) Series: 'a' [i64] [ 1 5 3 ] TrrN)rcollectrr _df) r r r r!rr"rrr#rlazys rassert_frame_equalr*!sp %dE 2 2D 6llnnemmooe{D))4 U+C+C%D  '-!-      rc d} t||} t||||||||| | rdnd} | d} t| #t$rYdSwxYw)ax Assert that the left and right frame are **not** equal. This function is intended for use in unit tests. .. versionchanged:: 0.20.31 The `check_dtype` parameter was renamed `check_dtypes`. .. versionchanged:: 1.32.3 The `rtol` and `atol` parameters were renamed to `rel_tol` and `abs_tol`, respectively. Parameters ---------- left The first DataFrame or LazyFrame to compare. right The second DataFrame or LazyFrame to compare. check_row_order Requires row order to match. check_column_order Requires column order to match. check_dtypes Requires data types to match. check_exact Requires float values to match exactly. If set to `False`, values are considered equal when within tolerance of each other (see `rel_tol` and `abs_tol`). Only affects columns with a Float data type. rel_tol Relative tolerance for inexact checking. Fraction of values in `right`. abs_tol Absolute tolerance for inexact checking. categorical_as_str Cast categorical columns to string before comparing. Enabling this helps compare columns that do not share the same string cache. See Also -------- assert_frame_equal assert_series_not_equal Examples -------- >>> from polars.testing import assert_frame_not_equal >>> df1 = pl.DataFrame({"a": [1, 2, 3]}) >>> df2 = pl.DataFrame({"a": [1, 2, 3]}) >>> assert_frame_not_equal(df1, df2) Traceback (most recent call last): ... AssertionError: DataFrames are equal (but are expected not to be) T) r r r!r rr"rrr# LazyFrames DataFramesz' are equal (but are expected not to be)N)rr*AssertionError) r r r r!rr"rrr#rr)objectsmsgs rassert_frame_not_equalr1sD %dE 2 2D"1+%#1 #'8,,LAAAS!!! sA AA)r r r r r r)r r r r r rr!rrrr"rrr$rr$r#rr r%) __future__r contextlibtypingrpolars._utils.deprecationrpolars.dataframerpolars.lazyframerpolars.testing.asserts.utilsrsuppress ImportError polars._plrr rr*r1rrr=sF""""""AAAAAA&&&&&&&&&&&&>>>>>>Z%%66555555666666666666666    $]NINNNVYAAAVYAAA !#$jjjjjBABAONjZ]NINNNVYAAAVYAAA !#$S"S"S"S"S"BABAONS"S"S"sA  AA