BVh@ jdZddlZddlZddlZddlZGddeZGddejeZ y)zRun doctests for tensorflow.NcveZdZdZej dj ddddejZdZ y ) _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_digitsdigitsexponentcg}g}t|jj|D]9\}}|dzdk(r|j| |jt |;|t j |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)selfstringtextsfloatsiparts T/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/tools/docs/tf_doctest_lib.py__call__z_FloatExtractor.__call__Hsr E FT^^11&9:#4 Q! T eDk" # "((6" ""N) __name__ __module__ __qualname____doc__recompileformatVERBOSEr rrrrrsM bjj * &";$*  + jjA )D#rrceZdZdZfdZej dZej dejZ d dZ dZ e jdZfdZfd ZxZS) TfDoctestOutputCheckerzACustomizes how `want` and `got` are compared, see `check_output`.cftt| |i|t|_d|_d|_y)N)superr%__init__rextract_floats text_goodfloat_size_good)rargskwargs __class__s rr(zTfDoctestOutputChecker.__init__fs2 $0$A&A)+DDNDrz\bat 0x[0-9a-f]*?>zc4tj||||S)N)rtolatol)rallclose)rwantgotr0r1s r _allclosez TfDoctestOutputChecker._allcloseqs ;;tSt$ 77rcH|jjd|}|||k7fS)Nz\1)_NUMPY_OUTPUT_REsub)rrmodified_strings r_tf_tensor_numpy_outputz.TfDoctestOutputChecker._tf_tensor_numpy_outputts*++//v>O Ov5 55ra ############################################################# Check the documentation (https://www.tensorflow.org/community/contribute/docs_ref) on how to write testable docstrings. #############################################################c|r|sy|d}||k(ry|jjd|}|j|\}}|r|j|\}}|j|\}|_|Dcgc]}|j d}}dj |}d|vrtjdd|}|j|\}|_tt|/||||_ |jsy |jjd k(ry|jj|jjk(|_|jr&|j|j|jSy cc}w) 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. Tzat ...> z...z....z \.\.\.\.+)r3r4 optionflagsFr) _ADDRESS_REr8r:r) want_floatsstripjoinr got_floatsr'r% check_outputr*sizer+r5) rr3r4r> want_changed_want_text_partsrwant_text_wildr.s rrDz#TfDoctestOutputChecker.check_output~sx> 4  | d s{       4 0D55d;D,++C0fc1)-(;(;D(A%OT% 4CC4tzz#COCZZ0N vvlE>Bn,,S1At14E +F?DN >>  ! ,,11T__5I5IID  ^^D,,doo >> 3Ds4E:c|g}|jr|js|jd|j|jdj |}t t ||||S)NzZ CAUTION: tf_doctest doesn't work if *some* of the *float output* is hidden with a "...". )r*r+rMESSAGErBr'r%output_difference)rexampler4r>r.s rrMz(TfDoctestOutputChecker.output_differencesj %C  ~~  ! ! > ?JJt|| ))C.C ( **13 DFr)MbP?rO)rrrrr(rr r?DOTALLr7r5r:textwrapdedentrLrDrM __classcell__)r.s@rr%r%csnI  01+ RZZ rXs=# H#fH#V}FW22F}Fr