M/Ph dZddlZddlmZmZmZmZmZmZddl Z ddl m Z e de jdDZee dre jndZ ddee jge jfd eeeed ee jgee jffd Z ddee jge jfd eeeed ee jgeffd Z ddee jge jfd eeeed ee jgeffd Zdee jge jfd ee jgeffdZdeed ee jegeffdZded ee jegeffdZdeeeefd ee jegeffdZdeeeefd ee jge jffdZdS)z/ A selection of testing utilities for visions. N)CallableDictListOptionalTypeUnion)series_handle_nullsc#4K|]}t|VdSN)int).0is b/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/visions/backends/pandas/test_utils.py r s(AA!s1vvAAAAAA.NAfn extra_errorsreturnctttg|r|t jdt jdtt jffd }|S)a7A coercion test evaluator Evaluates a coercion function and optionally returns the coerced series. Args: fn: A function coercing a Series to another Series. extra_errors: Additional exceptions to catch Returns: The coerced series if the coercion succeeds otherwise None. seriesrcN |S#t$rYdSwxYwr )tuple)r error_listrs rfz$option_coercion_evaluator..f!s@ 2f:: Z     44 s $$) ValueError TypeErrorAttributeErrorextend functoolswrapspdSeriesr)rrrrs` @roption_coercion_evaluatorr%si8J(,'''_R") 3 Hrct||tj|dtjdt ffd }|S)aA coercion test generator Creates a coercion test based on a provided coercion function. Args: fn: A function coercing a Series to another type. extra_errors: Additional exceptions to catch Returns: Whether the coercion failed or was successful. rrc&|}|dndS)NTFrresulttesters rrzcoercion_test..f:s)ttu4rr%r!r"r#r$boolrrrr+s @r coercion_testr/+s^'r< 8 8F_R5")5555555 Hrct||tjdtjdt ffd }|S)aA coercion equality test generator Creates a coercion test based on a provided coercion function which also enforces equality constraints on the output. This is useful when you want to change the data type of a series without necessarily changing the data, for example, when converting an integer to a float. Args: fn: A function coercing a Series to another type. extra_errors: Additional exceptions to catch Returns: Whether the coercion failed or was successful. rrcJ|}|dn|SNF)allr)s rrzcoercion_true_test..fSs&uuFJJLL8rr,r.s @rcoercion_true_testr4Bs^'r< 8 8F_V9")9999999 Hrct|tjdtjdt ffd }|S)aA coercion equality test generator Creates a coercion test based on a provided coercion function which also enforces equality constraints on the output. This is useful when you want to change the data type of a series without necessarily changing the data, for example, when converting an integer to a float. Args: fn: A function coercing a Series to another type. Returns: Whether the coercion failed or was successful. rrcp|}|dn&||Sr2)eqr3r)s rrz!coercion_equality_test..fjs6uuFIIf,=,=,A,A,C,CCrr,)rrr+s @rcoercion_equality_testr8[sd'r * *F_VD")DDDDDDD Hrmappingcftifdtjdtdtffd }|S)Nrstaterc<tfdDS)Nc3K|]J}t|VKdSr isinlistkeysr3)r single_maprs rrz6coercion_single_map_test..f..usMXX*6;;tJOO$5$56677;;==XXXXXXr)anyrr;r9s` rrz#coercion_single_map_test..fss(XXXXPWXXXXXXrr r#r$dictr-r9rs` rcoercion_single_map_testrHrs\+-YY")YDY$YYYYYY Hrcftifdtjdtdtffd }|S)Nrr;rc|tSr r>rDs rrz"coercion_multi_map_test..f{s1{{4 //0044666rrErGs` rcoercion_multi_map_testrKzsR+-77")7D7$777777 Hrct|trt|}n4t|trt |}nt d|S)aCreate a testing function for a single mapping or a list of mappings. Args: mapping: A dict with a mapping or a list of dicts Returns: Callable that checks if a series consists of the mappable values Examples: >>> coercion_map_test({"Yes": True, "No": False}) >>> coercion_map_test( >>> [ >>> {"Yes": True, "No": False}, >>> {"Y": True, "N": False}, >>> ] >>> ) 'Mapping should be dict or list of dicts) isinstancer@rHrFrKrrGs rcoercion_map_testrOs\$'4  D $W - - GT " "D #G , ,BCCC Hrcttr dDn$ttstddtjdtjffd }|S)zMaps a series given a mapping Args: mapping: a dict to map, or a list of dicts. Returns: A callable that maps the series. cHi|]}|D]\}}|| Sr()items)r dkvs r z coercion_map..s3???AQWWYY??TQ1a????rrMrrc.|Sr )map)rr9s rrzcoercion_map..fszz'"""r)rNr@rFrr#r$rGs` r coercion_maprYs'4  D?????  & &DBCCC#")# ###### Hrr )__doc__r!typingrrrrrrpandasr#$visions.backends.pandas.series_utilsr r __version__splitpandas_versionhasattrrpandas_na_valuer$ Exceptionr%r-r/r4r8rHrKrOrYr(rrrds">>>>>>>>>>>>>>>>DDDDDDAAr~';';C'@'@AAAAA"72t,,6"%%$ 59  ")bi'( 4Y01 ryk8BI../    :59  ")bi'( 4Y01 ryk4     259  ")bi'( 4Y01 ryk4     2 ")bi'(  ryk4     . d4j Xry$>OQU>U5V     T h 47H$7N.O     4:t# $  ry$%&    6 %T D 01 h {BI?U6V      r