o i e@sLdZddlZddlZddlZddlZGdddeZGdddejeZ dS)zRun doctests for Keras.Nc@s4eZdZdZedjdddddejZdd Z d S) _FloatExtractorzClass for extracting floats from a string. For example: >>> text_parts, floats = _FloatExtractor()("Text 1.0 Text") >>> text_parts ['Text ', ' Text'] >>> floats array([1.]) a ( # Captures the float value. (?: [-+]| # Start with a sign is okay anywhere. (?: # Otherwise: ^| # Start after the start of string (?<=[^\w.]) # Not after a word char, or a . ) ) (?: # Digits and exponent - something like: {digits_dot_maybe_digits}{exponent}?| # "1.0" "1." "1.0e3", "1.e3" {dot_digits}{exponent}?| # ".1" ".1e3" {digits}{exponent}| # "1e3" {digits}(?=j) # "300j" ) ) j? # Optional j for cplx numbers, not captured. (?= # Only accept the match if $| # * At the end of the string, or [^\w.] # * Next char is not a word char or "." ) z(?:[0-9]+\.(?:[0-9]*))z (?:\.[0-9]+)z (?:[0-9]+)z(?:[eE][-+]?[0-9]+))digits_dot_maybe_digits dot_digitsdigitsexponentcCsVg}g}t|j|D]\}}|ddkr||q |t|q |t|fS)aExtracts floats from a string. >>> text_parts, floats = _FloatExtractor()("Text 1.0 Text") >>> text_parts ['Text ', ' Text'] >>> floats array([1.]) Args: string: the string to extract floats from. Returns: A (string, array) pair, where `string` has each float replaced by "..." and `array` is a `float32` `numpy.array` containing the extracted floats. r) enumerate _FLOAT_REsplitappendfloatnparray)selfstringtextsfloatsipartrX/var/www/myenv/lib/python3.10/site-packages/keras/src/testing_infra/keras_doctest_lib.py__call__Js  z_FloatExtractor.__call__N) __name__ __module__ __qualname____doc__recompileformatVERBOSEr rrrrrrs   $rcsleZdZdZfddZedZedejZ dddZ d d Z e d Zfd d ZfddZZS)KerasDoctestOutputCheckerzACustomizes how `want` and `got` are compared, see `check_output`.cs*tj|i|t|_d|_d|_dS)N)super__init__rextract_floats text_goodfloat_size_good)rargskwargs __class__rrr"is z"KerasDoctestOutputChecker.__init__z\bat 0x[0-9a-f]*?>zMbP?cCstj||||dS)N)rtolatol)r allclose)rwantgotr+r,rrr _allclosetsz#KerasDoctestOutputChecker._allclosecCs|jd|}|||kfS)Nz\1)_NUMPY_OUTPUT_REsub)rrmodified_stringrrr_tf_tensor_numpy_outputws z1KerasDoctestOutputChecker._tf_tensor_numpy_outputz ############################################################# Check the documentation (go/testable-docstrings) on how to write testable docstrings. #############################################################cs|r|sdS|dur d}|jd|}||\}}|r#||\}}||\}|_d|}||\}|_tj|||d|_ |j sGdS|jj dkrOdS|jj |jj k|_ |j rc| |j|jSdS) aCompares the docstring output to the output gotten by running the code. Python addresses in the output are replaced with wildcards. Float values in the output compared as using `np.allclose`: * Float values are extracted from the text and replaced with wildcards. * The wildcard text is compared to the actual output. * The float values are compared using `np.allclose`. The method returns `True` if both the text comparison and the numeric comparison are successful. The numeric comparison will fail if either: * The wrong number of floats are found. * The float values are not within tolerence. Args: want: The output in the docstring. got: The output generated after running the snippet. optionflags: Flags passed to the doctest. Returns: A bool, indicating if the check was successful or not. TNzat ...>z...)r.r/ optionflagsFr) _ADDRESS_REr2r4r# want_floatsjoin got_floatsr! check_outputr$sizer%r0)rr.r/r6 want_changed_want_text_partswant_text_wildr(rrr;s,"  z&KerasDoctestOutputChecker.check_outputcsB|g}|jr|js|d||jd|}t|||S)NzZ CAUTION: tf_doctest doesn't work if *some* of the *float output* is hidden with a "...". )r$r%r MESSAGEr9r!output_difference)rexampler/r6r(rrrCs  z+KerasDoctestOutputChecker.output_difference)r*r*)rrrrr"rrr7DOTALLr1r0r4textwrapdedentrBr;rC __classcell__rrr(rr fs    Lr ) rdoctestrrFnumpyr objectr OutputCheckerr rrrrsN