q-Ph3 ddlmZddlmZmZddlmZddlmZm Z m Z m Z m Z m Z ddlmZddlmZmZmZddlmZddlmZerdd lmZd6dZedddddddddddd7d"Zd8d#Zd9d%Zd:d&Zd;d'Zd;d(Z dd1Z&d?d3Z'edddddddddddd7d4Z(d5S)@) annotations) TYPE_CHECKINGAny)deprecate_renamed_parameter)Array CategoricalListStringStruct unpack_dtypes) FLOAT_DTYPES) ComputeErrorInvalidOperationError ShapeError)Series)raise_assertion_error)DataTypeleftrrightreturnboolcd}t|trt|ts6tddt|jt|jdS)NTinputszunexpected input types) isinstancerrtype__name__)rr__tracebackhide__s ]/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/polars/testing/asserts/series.py_assert_correct_input_typersb tV $ $ E6)B)B   $ JJ  KK    4 check_dtype check_dtypesz0.20.31)versionTFgh㈵>g:0yE>)r" check_names check_order check_exactrtolatolcategorical_as_strrr$r%r&r'floatr(r)Nonec d} t||||kr6tdd|||r,|j|jkrtdd|j|j|r,|j|jkrtdd|j|jt |||||||dS)a[ Assert that the left and right Series are equal. Raises a detailed `AssertionError` if the Series 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 Series to compare. right The second Series to compare. check_dtypes Requires data types to match. check_names Requires names to match. check_order Requires elements to appear in the same order. 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, given as a fraction of the 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 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_series_equal >>> s1 = pl.Series([1, 2, 3]) >>> s2 = pl.Series([1, 5, 3]) >>> assert_series_equal(s1, s2) Traceback (most recent call last): ... AssertionError: Series are different (exact value mismatch) [left]: [1, 2, 3] [right]: [1, 5, 3] Trzlength mismatchz name mismatchzdtype mismatchr%r&r'r(r)N)rlenrnamedtype_assert_series_values_equal) rrr"r$r%r&r'r(r)rs rassert_series_equalr2$sNtU+++ xxzzUYY[[  h(9488::uyy{{SSSPtyEJ..hEJOOOS ek11h(8$*ekRRR    -r cd}|rt|}t|}|st||\}} ||}n[#t$r(} t dd|j|j| Yd} ~ n.d} ~ wt $r} t dd||| Yd} ~ nd} ~ wwxYwt|j|jr t| || |||||dS#t$rB} t dd| | | Yd} ~ nd} ~ wwxYw| sdS|s2|j r|j s7t dd | | t||t||t!||||| dS) z0Assert that the values in both Series are equal.Trzincompatible data types)rrcauseNzincompatible lengths)rrr&r'r(r)nested value mismatchzexact value mismatchrr)r'r()_categorical_series_to_string _sort_series ne_missingrrr0r_comparing_nested_floats"_assert_series_nested_values_equalfilterAssertionErrorto_listanyis_float _assert_series_null_values_match_assert_series_nan_values_match&_assert_series_values_within_tolerance) rrr%r&r'r(r)runequalexcs rr1r1s5,T22-e44 0"4// e //%((      %+                "             EK88  .[[))ll7++'#5     " F    !'\\^^mmoo            ;;== $*--// u{7K7K7M7M  ,4<<>>    %T5111#D%000*     s;A B'A<< B' B""B';D E 8E  Etuple[Series, Series]c |}|}n$#t$r}d}t||d}~wwxYw||fS)NzBcannot set `check_order=False` on Series with unsortable data type)sortr TypeError)rrrEmsgs rr8r8sa&yy{{  &&&Rnn#%& ;s(+ A AA c d}t|j|jrDt||D]1\}}||tdd||t ||d||||2dS|j|j} } t| | D]\}}t ||d||||dS)NTrr5r-)_comparing_listsr0ziprr1structunnest) rrr&r'r(r)rs1s2lsrss rr;r;s EK00$&&  FBzRZ%h0GRPPP ' '#5        ##%%u|':':'<')rrrnull_value_mismatchs rrArA sw,,..EMMOO;    +T\\^^U]]__       r c.d}t|j|jsdS||k}|r8t dd||dSdS)NTrznan value mismatch)_comparing_floatsr0is_nanr?rr>)rrrnan_value_mismatchs rrBrBs TZ 5 5%,,..8   LLNN MMOO        r rcR|o|SN)r@r6s rrXrX"s ==?? /u~~///r cB|ttfvo|ttfvSr\)r rr6s rrLrL&s D%= ;UtUm%;;r c.|tko |tkSr\)r r6s r_comparing_structsr_*s 6> -evo-r ct||st||sdSttt |zo#ttt |zS)NF)rLr_rr r r6s rr:r:.si T5 ) )-?e-L-Lu  }T222 3 3 }U+++99r rDcd}||||}}||z }|||zz} || k|z||kz} | s8t dd||dSdS)NTrzvalue mismatch)r<abs is_finiteallrr>) rrrDr'r(r left_unequal right_unequal difference tolerancewithin_tolerances rrCrC7s"&++g"6"6 W8M8M-L.3355Jtm//1111I"i/=3J3J3L3LL %    ! !    LLNN MMOO        r sch|j}t|}||kr||}|S)z?Cast a (possibly nested) Categorical Series to a String Series.)r0"_categorical_dtype_to_string_dtypecast)rjr0 noncat_dtypes rr7r7Rs7 GE5e<z6_categorical_dtype_to_string_dtype..fs7    F6qw??   r ) rrr r rlinnerrsizer fields)r0 inner_castrws rrlrl[s%%%xx E4 7 DD J E5 ! ! 7 DD Z,,, E6 " "  \   f~~ r c d} t|| t||||||||| d} t| #t$rYdSwxYw)a Assert that the left and right Series 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 Series to compare. right The second Series to compare. check_dtypes Requires data types to match. check_names Requires names to match. check_order Requires elements to appear in the same order. 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, given as a fraction of the 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 Examples -------- >>> from polars.testing import assert_series_not_equal >>> s1 = pl.Series([1, 2, 3]) >>> s2 = pl.Series([1, 2, 3]) >>> assert_series_not_equal(s1, s2) Traceback (most recent call last): ... AssertionError: Series are equal (but are expected not to be) T) rrr"r$r%r&r'r(r)z-Series are equal (but are expected not to be)N)rr2r=) rrr"r$r%r&r'r(r)rrJs rassert_series_not_equalrzosztU+++"%###1 >S!!! s= A  A N)rrrrrr)rrrrr"rr$rr%rr&rr'r*r(r*r)rrr+)rrrrr%rr&rr'r*r(r*r)rrr+)rrrrrrF)rrrrr&rr'r*r(r*r)rrr+)rrrrrr+)rrrrrr) rrrrrDrr'r*r(r*rr+)rjrrr)r0rrr)) __future__rtypingrrpolars._utils.deprecationrpolars.datatypesrrr r r r polars.datatypes.groupr polars.exceptionsrrr polars.seriesrpolars.testing.asserts.utilsrpolarsrrr2r1r8r;rArBrXrLr_r:rCr7rlrzrqr rrs""""""%%%%%%%%AAAAAA0/////MMMMMMMMMM >>>>>>     ]NINNN $[[[[[ON[|QQQQh''''T         0000<<<<....    6    (]NINNN $O"O"O"O"O"ONO"O"O"r