ˀhi# XddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZed d d d d d ddddd ddZ ddZ ddZddZed d d d d d ddddd ddZy)) annotations)cast)deprecate_renamed_parameter) DataFrame)InvalidOperationError) LazyFrame)_assert_series_values_equal)raise_assertion_error check_dtype check_dtypesz0.20.31)versionTFgh㈵>g:0yE>)check_row_ordercheck_column_orderr check_exactrtolatolcategorical_as_strc ~d} t||} | rdnd} t||||| | r |j|j}}tt|tt|}}|j |j k7r"t | d|j |j |st||\}}|jD]7} |j| |j| }} t| |d||||9y #t$r9}t | d| | j|j|Yd }~xd }~wwxYw) 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`. 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 `rtol` and `atol`). Only affects columns with a Float data type. rtol Relative tolerance for inexact checking. Fraction of values in `right`. atol 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]: [1, 2, 3] [right]: [1, 5, 3] T LazyFrames DataFrames)rr objectsznumber of rows does not match) check_orderrrrrzvalue mismatch for column )causeN) _assert_correct_input_type_assert_frame_schema_equalcollectrrheightr _sort_dataframescolumns get_columnr AssertionErrorto_list)leftrightrrr rrrr__tracebackhide__lazyrcs_lefts_rightexcs x/var/www/html/wine-match-dev/backend/winematch-backend/venv/lib/python3.12/site-packages/polars/testing/asserts/frame.pyassert_frame_equalr, s<L %dE 2D"l G  -!  llnemmoey$'i)?%D {{ell" 4dkk5<<  &tU3 e \\//!,e.>.>q.A  ' '#5   !,QE2 !    s%C:: D</D77D<cd}t|trt|tryt|trt|trytddt |j t |j y)NTFinputszunexpected input types) isinstancerrr type__name__)r#r$r%s r+rrs^$ "z%'C D) $E9)E  $ J   K  cd}|j|j}}||k(ry|j|jk7rY|Dcgc] }||vs| c}x} rd| d|ddd} t| |Dcgc] }||vs| } }d| d|ddd} t| |r+t|t|} } | | k7rd}t ||| | |r/t |t |}}|s||k7rd }t ||||yyycc}wcc}w) NTzcolumns z in left z, but not in rightz in right z, but not in leftz!columns are not in the same orderzdtypes do not match)collect_schemakeysr!listr dict)r#r$r rrr% left_schema right_schemar'left_not_rightmsgright_not_left left_columns right_columnsdetailleft_schema_dictright_schema_dicts r+rrsT $ 3 3 5u7K7K7MKl"\..00)4NA8MaN N> N^.i ~EWXC % %)5NA+9MaNNN^.j"FWXC % %&*;&7l9Km = (8F !'6< O.2;.?lAS+ !15F!F*F !'63CEV W"GOOs DD< DDc|j} |j|}|j|}||fS#t$r}d}t||d}~wwxYw)NzCcannot set `check_row_order=False` on frame with unsortable columns)rsortr TypeError)r#r$byr*r<s r+rrsY B&yy} 2 ; !&Sn#%&s"4 AA  Ac d} t||} t||||||||| | rdnd} | d} t| #t$rYywxYw)a 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`. 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 `rtol` and `atol`). Only affects columns with a Float data type. rtol Relative tolerance for inexact checking. Fraction of values in `right`. atol 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$rrr rrrrrrz' are equal (but are expected not to be)N)rr,r!) r#r$rrr rrrrr%r&rr<s r+assert_frame_not_equalrHsvx %dE 2D"1+%#1 #',L @AS!! s: AAN)r#DataFrame | LazyFramer$rIrboolrrJr rJrrJrfloatrrKrrJreturnNone)r#rIr$rIrLrJ) r#rIr$rIr rJrrJrstrrLrM)r#rr$rrLztuple[DataFrame, DataFrame]) __future__rtypingrpolars._utils.deprecationrpolars.dataframerpolars.exceptionsrpolars.lazyframerpolars.testing.asserts.seriesr polars.testing.asserts.utilsr r,rrrrHr2r+rXs"A&3&E>]NIN !#$q q q q  q  qq q qq qOqh  (=   $$X $X $X $X  $X  $X $XN]NIN !#$O" O" O" O"  O"  O"O" O" O"O" O"OO"r2