o i e@ @sLdZddlZddlZddlZddlZGdddeZGdddejeZ dS)zRun doctests for tensorflow.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 np.array([1.0]) 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 np.array([1.0]) 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)selfstringtextsfloatsipartrS/var/www/myenv/lib/python3.10/site-packages/tensorflow/tools/docs/tf_doctest_lib.py__call__Hs  z_FloatExtractor.__call__N) __name__ __module__ __qualname____doc__recompileformatVERBOSEr rrrrrrs  "rcsleZdZdZfddZedZedejZ dddZ d d Z e d Zfd d ZfddZZS)TfDoctestOutputCheckerzACustomizes how `want` and `got` are compared, see `check_output`.cs.tt|j|i|t|_d|_d|_dS)N)superr __init__rextract_floats text_goodfloat_size_good)rargskwargs __class__rrr"fs zTfDoctestOutputChecker.__init__z\bat 0x[0-9a-f]*?>zMbP?cCstj||||dS)N)rtolatol)r allclose)rwantgotr+r,rrr _allcloseqsz TfDoctestOutputChecker._allclosecCs|jd|}|||kfS)Nz\1)_NUMPY_OUTPUT_REsub)rrmodified_stringrrr_tf_tensor_numpy_outputts z.TfDoctestOutputChecker._tf_tensor_numpy_outputa ############################################################# Check the documentation (https://www.tensorflow.org/community/contribute/docs_ref) on how to write testable docstrings. #############################################################cs|r|sdS|dur d}||krdS|jd|}||\}}|r)||\}}||\}|_dd|D}d|}d|vrHtd d|}||\}|_tt |j |||d |_ |j sad S|jj d kridS|jj |jj k|_ |j r}||j|jSd S) a{Compares 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 ...>cSsg|]}|dqS) )strip).0rrrr sz7TfDoctestOutputChecker.check_output..z...z....z \.\.\.\.+)r.r/ optionflagsFr) _ADDRESS_REr2r4r# want_floatsjoinr got_floatsr!r check_outputr$sizer%r0)rr.r/r: want_changed_want_text_partswant_text_wildr(rrr?~s6   z#TfDoctestOutputChecker.check_outputcsF|g}|jr|js|d||jd|}tt||||S)NzZ CAUTION: tf_doctest doesn't work if *some* of the *float output* is hidden with a "...". )r$r%r MESSAGEr=r!r output_difference)rexampler/r:r(rrrGs    z(TfDoctestOutputChecker.output_difference)r*r*)rrrrr"rrr;DOTALLr1r0r4textwrapdedentrFr?rG __classcell__rrr(rr cs     Sr ) rdoctestrrJnumpyr objectr OutputCheckerr rrrrsK