q-Phi# ddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZed d d d d d dddddd)dZd*d Zd+d#Zd,d&Zed d d d d d dddddd)d'Zd(S)-) 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_strleftDataFrame | LazyFramerightrboolrrrfloatrrreturnNonec d} t||} | rdnd} t||||| | r(||}}tt|tt|}}|j|jkrt | d|j|j|st||\}}|jD]} | | | | }} t| |d||||D#t$rE}t | d| | | |Yd }~d }~wwxYwd S) 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)rrrrr rrrr__tracebackhide__lazyrcs_lefts_rightexcs \/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/polars/testing/asserts/frame.pyassert_frame_equalr1 sL %dE 2 2D"4ll G  -!  6llnnemmooey$''i)?)?%D {el"" 4dk5<    4&tU33 e \//!,,e.>.>q.A.A  ' '#5         !2Q22  !!           s6D  E;EEc&d}t|trt|trdSt|trt|trdStddt |jt |jdS)NTFinputszunexpected input types) isinstancerrr type__name__)rrr*s r0r!r!s$ ""  z%'C'C  u D) $ $ E9)E)E t  $ JJ  KK      rstrctd}||ckrdSkr^fdDx}r d|d|ddd}t|fdD}d|d|ddd }t||r8tt} } | | krd } t || | | |rz._assert_frame_schema_equal..s#NNNA8M8Ma8M8M8Mr7zcolumns z in left z, but not in rightcg|]}|v| Sr;r;)r<r, left_schemas r0r>z._assert_frame_schema_equal..s#NNNA+9M9Ma9M9M9Mr7z in right z, but not in leftz!columns are not in the same orderzdtypes do not match)collect_schemakeysr(listr dict)rrr rrr*left_not_rightmsgright_not_left left_columns right_columnsdetailleft_schema_dictright_schema_dictrAr=s @@r0r"r"s $ 3 3 5 5u7K7K7M7MKl""\..0000NNNNNNN N> &X^XX XXXC %% %NNNNNNNNX^XX"XXXC %% %P&*;&7&7l9K9Km = ( (8F !'6< O O OX.2;.?.?lASAS+  X!15F!F!F*F !'63CEV W W W W W XX!F!Fr7rtuple[DataFrame, DataFrame]c|j} ||}||}n$#t$r}d}t||d}~wwxYw||fS)NzCcannot set `check_row_order=False` on frame with unsortable columns)r&sortr TypeError)rrbyr/rGs r0r%r%sl B&yy}} 2 &&&Snn#%& ;s*4 AAAc d} t||} t||||||||| | rdnd} | d} t| #t$rYdSwxYw)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) rrrrr rrrrrrz' are equal (but are expected not to be)N)r!r1r() rrrrr rrrrr*r+rrGs r0assert_frame_not_equalrTsx %dE 2 2D"1+%#1 #'8,,LAAAS!!! sA AAN)rrrrrrrrr rrrrrrrrrrr)rrrrrr) rrrrr rrrrr8rr)rrrrrrN) __future__rtypingrpolars._utils.deprecationrpolars.dataframerpolars.exceptionsrpolars.lazyframerpolars.testing.asserts.seriesr polars.testing.asserts.utilsr r1r!r"r%rTr;r7r0r]s""""""AAAAAA&&&&&&333333&&&&&&EEEEEE>>>>>>]NINNN !#$qqqqqONqh    $$X$X$X$XN]NINNN !#$O"O"O"O"O"ONO"O"O"r7