o $&]in+@s4UddlmZddlmZddlmZmZmZddlm Z m Z ddl m Z ddl mZmZmZmZmZmZddlmZddlmZergdd lmZdd lmZdd lmZmZeeeeegd fZd e d<ddddddddd;d"d#Z!dd*d+Z$d?d,d-Z%d@d4d5Z&dAd7d8Z'dBd9d:Z(d S)C) annotations)partial) TYPE_CHECKINGAnyCallable)qualified_type_name zip_strictis_narwhals_series)ArrayBoolean CategoricalListStringStruct) new_series)raise_series_assertion_error) TypeAlias)Series) IntoSeriesTSeriesTNrCheckFnTFgh㈵>g:0yE> check_dtypes check_names check_order check_exactrel_tolabs_tolcategorical_as_strleftSeries[IntoSeriesT]rightrboolrrrrfloatrrreturnNonec Csd} tdd||fDr"dtt|dtt|d} t| t||||d|s@|jr7d} t| ||}}t ||\} } |sN|j s[t | | |||||d d St | | ||d d S) u"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. Arguments: 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 `rel_tol` and `abs_tol`). Only affects columns with a Float data type. rel_tol: Relative tolerance for inexact checking, given as a fraction of the 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. Examples: >>> import pandas as pd >>> import narwhals as nw >>> from narwhals.testing import assert_series_equal >>> s1 = nw.from_native(pd.Series([1, 2, 3]), series_only=True) >>> s2 = nw.from_native(pd.Series([1, 5, 3]), series_only=True) >>> assert_series_equal(s1, s2) # doctest: +ELLIPSIS Traceback (most recent call last): ... AssertionError: Series are different (exact value mismatch) [left]: ┌───────────────┐ |Narwhals Series| |---------------| | 0 1 | | 1 2 | | 2 3 | | dtype: int64 | └───────────────┘ [right]: ┌───────────────┐ |Narwhals Series| |---------------| | 0 1 | | 1 5 | | 2 3 | | dtype: int64 | └───────────────┘ Tcss|]}t| VqdS)Nr ).0objr)]/var/www/html/IGF-ODF-V3/venv/lib/python3.10/site-packages/narwhals/testing/asserts/series.py Tsz&assert_series_equal..z4Expected `narwhals.Series` instance, found: [left]: z [right]: zp Hint: Use `nw.from_native(obj, series_only=True) to convert each native object into a `narwhals.Series` first.)rrzA`check_order=False` is not supported (yet) with nested data type.)rrrrr)rrN) anyrtype TypeError_check_metadatadtype is_nestedNotImplementedErrorsort_check_null_valuesis_float_check_exact_values_check_approximate_values) r r"rrrrrrr__tracebackhide__msg left_vals right_valsr)r)r*assert_series_equals6=     r<rc Cs|j|j}}||krtd||t|t|}}||kr$td|||j|j}} |r7|| kr7td|| |j|j} } |rL| | krNtd| | dSdSdS)zECheck metadata information: implementation, length, dtype, and names.zimplementation mismatchzlength mismatchzdtype mismatchz name mismatchN)implementationrlenr0name) r r"rr left_impl right_implleft_len right_len left_dtype right_dtype left_name right_namer)r)r*r/vs     r/tuple[SeriesT, SeriesT]cCs\||}}||}}||ks||kr"td||||||fS)z8Check null value consistency and return non-null values.znull value mismatch) null_countis_nullr,rfilter)r r"left_null_countright_null_countleft_null_maskright_null_maskr)r)r*r4sr4c CsF|j}|j|j}} |r|j|dddd} n|t|ttfrKt| ttfrK|| krKtt|dd||||d} t |||| | dt ddgt |d} nJt|t rut| t ruttddd||||d} t |||| | dt ddgt |d} n t|trt| trt|||d } t d| gt |d} n||k} | rtd ||d Sd S) z2Check exact value equality for various data types.rTrr nans_equalFr)check_fn)r0backend)rexact value mismatchN)r=r0 is_numericis_close isinstancer rrr<_check_list_likerr r _check_structr _check_categoricalr,r) r r"rrrrrr@rDrEis_not_equal_maskrR _not_equalr)r)r*r6sZ      r6cCs<|j|||dd}|rtd||||dSdS)z0Check approximate value equality with tolerance.TrPzvalues not within toleranceN)rWr,rrK)r r"rris_not_close_maskr)r)r*r7sr7r:r;rD List | ArrayrErRc Csh|j}z t||D]\}}|td||j|dtd||j|dq WdSty3td||YdSw)NrS)valuesr0rTznested value mismatch)r=rrinnerAssertionErrorr)r:r;rDrErRimplleft_val right_valr)r)r*rYs  rYrcCs^zt|j|jD]\}}||j|j|j|jqWdSty.td||YdSw)NrU)rfieldsstructfieldr?rbr)r:r;rDrErR left_field right_fieldr)r)r*rZs    rZc CsT|r|t|t}}z||kWSty)}zd}t||d}~ww)zTry to compare if any element of categorical series' differ. Inability to compare means that the encoding is different, and an exception is raised. z:Cannot compare categoricals coming from different sources.N)castrr, Exceptionrb)r:r;rexcr9r)r)r*r[s r[)r r!r"r!rr#rr#rr#rr#rr$rr$rr#r%r&) r rr"rrr#rr#r%r&)r rr"rr%rH)r rr"rrr#rr#rr$rr$rr#r%r&) r rr"rrr$rr$r%r&) r:rr;rrDr_rEr_rRrr%r&) r:rr;rrDrrErrRrr%r&)r:rr;rrr#r%r#)) __future__r functoolsrtypingrrrnarwhals._utilsrrnarwhals.dependenciesr narwhals.dtypesr r r rrrnarwhals.functionsrnarwhals.testing.asserts.utilsrtyping_extensionsrnarwhals.seriesrnarwhals.typingrrr__annotations__r<r/r4r6r7rYrZr[r)r)r)r*s8         a  A