M/Ph$ ,dZddlZddlmZmZmZmZmZmZddl Z ddl m Z  ddee j ge j fdeeeedee j gee j ffdZ ddee j ge j fdeeeedee j geffd Z ddee j ge j fdeeeedee j geffd Zdee j ge j fdee j geffd Zd eedee j egeffd Zd edee j egeffdZd eeeefdee j egeffdZd eeeefdee j ge j ffdZdS)z/ A selection of testing utilities for visions. N)CallableDictListOptionalTypeUnion)array_handle_nullsfn extra_errorsreturnctttg|r|t jdt jdtt jffd }|S)a3A coercion test evaluator Evaluates a coercion function and optionally returns the coerced array. Args: fn: A function coercing a array to another array. extra_errors: Additional exceptions to catch Returns: The coerced array if the coercion succeeds otherwise None. arrayr cN |S#t$rYdSwxYwN)tuple)r error_listr s a/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/visions/backends/numpy/test_utils.pyfz$option_coercion_evaluator..fs@ 2e99 Z     44 s $$) ValueError TypeErrorAttributeErrorextend functoolswrapsnpndarrayr)r r rrs` @roption_coercion_evaluatorr si8J(,'''_R 4 Hct||tj|dtjdt ffd }|S)aA coercion test generator Creates a coercion test based on a provided coercion function. Args: fn: A function coercing a array to another type. extra_errors: Additional exceptions to catch Returns: Whether the coercion failed or was successful. rr c&|}|dndS)NTFrresulttesters rrzcoercion_test..f7s)ttu4rrrrrrboolr r rr$s @r coercion_testr((s^'r< 8 8F_R55555555 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 array without necessarily changing the data, for example, when converting an integer to a float. Args: fn: A function coercing a array to another type. extra_errors: Additional exceptions to catch Returns: Whether the coercion failed or was successful. rr cJ|}|dn|SNF)allr"s rrzcoercion_true_test..fPs&uuEIIKK7rr%r's @rcoercion_true_testr-?s^'r< 8 8F_V88888888 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 array without necessarily changing the data, for example, when converting an integer to a float. Args: fn: A function coercing a array to another type. Returns: Whether the coercion failed or was successful. rr cL|}|dntj||Sr+)r array_equalr"s rrz!coercion_equality_test..fgs*uuBN5&,I,IIrr%)r rr$s @rcoercion_equality_testr1Xsd'r * *F_VJJJJJJJJ Hrmappingcftifdtjdtdtffd }|S)Nrstater c<tfdDS)Nc3K|]J}tjt|VKdSrrisinlistkeysr,).0 single_maprs r z6coercion_single_map_test..f..rs^  >HBGE4  1 122 3 3 7 7 9 9      r)anyrr4r2s` rrz#coercion_single_map_test..fps<    LS      rr rrdictr&r2rs` rcoercion_single_map_testrCosR+-   D $      Hrcftifdtjdtdtffd }|S)Nrr4r ctj|tSrr7r?s rrz"coercion_multi_map_test..fzs1wud7<<>>223377999rr@rBs` rcoercion_multi_map_testrFysR+-:::D:$:::::: 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 array 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) isinstancer9rCrArFrrBs rcoercion_map_testrJs\$'4  D $W - - GT " "D #G , ,BCCC Hrcttr dDn$ttstdt jfd}|S)zMaps a array given a mapping Args: mapping: a dict to map, or a list of dicts. Returns: A callable that maps the array. cHi|]}|D]\}}|| Sr!)items)r;dkvs r z coercion_map..s3???AQWWYY??TQ1a????rrHcD|tjSr)getrnan)valuer2s rzcoercion_map..s7;;ubf#=#=r)rIr9rArr vectorizerBs` r coercion_maprXsm'4  D?????  & &DBCCC ====>>A Hrr)__doc__rtypingrrrrrrnumpyr"visions.backends.numpy.array_utilsr r Exceptionrr&r(r-r1rCrFrJrXr!rrr^s>>>>>>>>>>>>>>>>AAAAAA 59  "*rz)* 4Y01 rzlHRZ001    :59  "*rz)* 4Y01 rzlD !    259  "*rz)* 4Y01 rzlD !    2 "*rz)*  rzlD !    . d4j Xrz4>PRV>V5W     T h D7I47O.P     4:t# $  rz4 $&'    6 4:t# $  rzlBJ&'      r