`ij dZddlZddlZddlmZmZddlmZm Z m Z m Z m Z d,dZ d,dZd,dZd,d Zejd gd d ed eddfdZejdgdd ededededdf dZejdgdd edejedejedededdf dZd,dZd,dZd,dZd,dZd,dZd,dZd,d Zd,d!Z d,d"Z!d,d#Z"d,d$Z#d,d%Z$ejd gd&d ed eddfd'Z%ejd d(gd ed eddfd)Z&ejd d*gd ed eddfd+Z'dS)-z*Tests for Google-style docstring routines.N) ParseErrorRenderingStyle) GoogleParserSection SectionTypecomposeparsereturnct}|d}|jdksJ|jdksJt |jdksJdS)zQTest parsing an unknown section with default GoogleParser configuration. z. Unknown: spam: a zUnknown:zspam: arN)rr short_descriptionlong_descriptionlenmetaparser docstrings v/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/docstring_parser/tests/test_google.py"test_google_parser_unknown_sectionrsp^^F  I  &* 4 4 4 4  % 2 2 2 2 y~  ! # # # # # #c|t}|d}|jdjdksJdS)zBTest parsing a multi-line parameter type with default GoogleParseraDescription of the function. Args: output_type (Literal["searchResults", "sourcedAnswer", "structured"]): The type of output. This can be one of the following: - "searchResults": Represents the search results. - "sourcedAnswer": Represents a sourced answer. - "structured": Represents a structured output format. Returns: bool: Indicates success or failure. r output_typeN)rr paramsarg_namers r,test_google_parser_multi_line_parameter_typer sG ^^F  I  A  '= 8 8 8 8 8 8rc ttddtjtddtjtddtjtddtjgd }|d }|jJ|jJt|j d ksJ|j dj dgksJ|j dj dksJ|j dj ddgksJ|j dj dksJ|j dj ddgksJ|j dj dksJ|j dj ddgksJ|j dj dksJ|j dj ddgksJ|j dj dksJ|j dj dgksJ|j dj dksJd S)zPTest parsing an unknown section with custom GoogleParser configuration. DESCRIPTIONdesc ARGUMENTSparam ATTRIBUTES attributeEXAMPLESexamplesF title_colona/ DESCRIPTION This is the description. ARGUMENTS arg1: first arg arg2: second arg ATTRIBUTES attr1: first attribute attr2: second attribute EXAMPLES Many examples More examples NrzThis is the description.arg1z first argarg2z second argattr1zfirst attributeattr2zsecond attributezMany examples More examples) rrrSINGULARMULTIPLEr r r rrargs descriptionrs r"test_google_parser_custom_sectionsr46s% M6;+? @ @ K+*> ? ? L+{/C D D J K,@ A A  F  I&  & . . .  % - - - y~  ! # # # # >!  !fX - - - - >!  (,F F F F F >!  !gv%6 6 6 6 6 >!  (K 7 7 7 7 >!  !gv%6 6 6 6 6 >!  (L 8 8 8 8 >!  !k7%; ; ; ; ; >!  (,= = = = = >!  !k7%; ; ; ; ; >!  (,> > > > > >!  !j\ 1 1 1 1 >!  (,J J J J J J Jrctd}|tddtj|d}|jdksJ|jdksJ|d}|jdksJ|jd ksJ|d }t|j d ksJ|j d j dgksJ|j d j d ksJdS)zjTest parsing an unknown section with custom GoogleParser configuration that was set at a runtime. Fr$NotenotezE short description Note: a note zshort descriptionzNote: a notez8 short description Note a note z Note a notezD short description Note a note r'rza noteN) r add_sectionrrr0r r r rrr2r3rs r(test_google_parser_custom_sections_afterr9gs1e , , ,F wvv{/CDDEEE  I  &*= = = = =  %)< < < < <  I  &*= = = = =  % 6 6 6 6  I y~  ! # # # # >!  !fX - - - - >!  (H 4 4 4 4 4 4rzsource, expected))N) NShort descriptionr=z Short description r=z Short description r=sourceexpectedcbt|}|j|ksJ|jJ|jrJdS)zTest parsing short description.N)r r r r)r@rArs rtest_short_descriptionrCsFf I  &( 2 2 2 2  % - - -~ rz?source, expected_short_desc, expected_long_desc, expected_blank))#Short description Long descriptionr=Long descriptionT)I Short description Long description r=rET)a Short description Long description Second line r=Long description Second lineT)"Short description Long descriptionr=rEF)H Short description Long description r=rEF)$ Short description Long description r=rEF)` Short description Long description Second line r=rHFexpected_short_descexpected_long_descexpected_blankct|}|j|ksJ|j|ksJ|j|ksJ|jrJdS)zTest parsing long description.N)r r r blank_after_short_descriptionr)r@rMrNrOrs rtest_long_descriptionrRsbLf I  &*= = = = =  %); ; ; ; ;  2n D D D D~ rzdsource, expected_short_desc, expected_long_desc, expected_blank_short_desc, expected_blank_long_desc))zR Short description Args: asd: r=NFF)zo Short description Long description Args: asd: r=rEFF)z Short description First line Second line Args: asd: r=First line Second lineFF)z Short description First line Second line Args: asd: r=rSTF)z Short description First line Second line Args: asd: r=rSTT)z4 Args: asd: NNFFexpected_blank_short_descexpected_blank_long_descct|}|j|ksJ|j|ksJ|j|ksJ|j|ksJt |jdksJdS)z2Test parsing newlines around description sections.r'N)r r r rQblank_after_long_descriptionrr)r@rMrNrTrUrs rtest_meta_newlinesrXsrf I  &*= = = = =  %); ; ; ; ;  26O O O O O  15M M M M M y~  ! # # # # # #rctd}|jdksJt|jdksJ|jdjddgksJ|jdjdksJ|jdjdksJdS) z*Test parsing multiline meta documentation.z Short description Args: spam: asd 1 2 3 r=r'rrspam asd 1 2 3N)r r rrr2rr3rs r$test_meta_with_multiline_descriptionr]Os   I  &*= = = = = y~  ! # # # # >!  !gv%6 6 6 6 6 >!  % / / / / >!  (,> > > > > > >rctd}|Jt|jdksJ|jd}|jdksJ|jsJ|jdksJ|jdksJ|jdksJdS) zTest parsing default arguments.aA sample function A function the demonstrates docstrings Args: arg1 (int): The firsty arg arg2 (str): The second arg arg3 (float, optional): The third arg. Defaults to 1.0. arg4 (Optional[Dict[str, Any]], optional): The last arg. Defaults to None. arg5 (str, optional): The fifth arg. Defaults to DEFAULT_ARG5. Returns: Mapping[str, Any]: The args packed in a mapping Nr/r+arg4zOptional[Dict[str, Any]]NonezThe last arg. Defaults to None.)r rrr is_optional type_namedefaultr3)rr_s rtest_default_argsrdcs I  y A % % % %  A D =F " " " "    >7 7 7 7 7 <6 ! ! ! !  @ @ @ @ @ @ @rc.td}|jdksJt|jdksJ|jdjddgksJ|jdjdksJ|jdjdksJ|jdjd d gksJ|jdjd ksJ|jdjd ksJ|jd jd d gksJ|jd jd ksJ|jd jdksJdS)zTest parsing multiple meta.z Short description Args: spam: asd 1 2 3 Raises: bla: herp yay: derp r=r+rrrZr[r'raisesblaherpr)yayderpN)r r rrr2rr3rbr\s rtest_multiple_metarksQ I  &*= = = = = y~  ! # # # # >!  !gv%6 6 6 6 6 >!  % / / / / >!  (,> > > > > >!  !h%6 6 6 6 6 >!  &% / / / / >!  (F 2 2 2 2 >!  !h%6 6 6 6 6 >!  &% / / / / >!  (F 2 2 2 2 2 2rctd}t|jdksJtd}t|jdksJ|jdjdksJ|jdjJ|jdjdksJ|jdjrJ|jdjd ksJ|jdjd ksJ|jdjd ksJ|jdjrJ|jd jd ksJ|jd jdksJ|jd jdksJ|jd jsJ|jdjdksJ|jdjdksJ|jdjdksJ|jdjsJtd}t|jd ksJ|jdjdksJ|jdjJ|jdjdksJ|jdjd ksJ|jdjd ksJ|jdjd ksJdS)zTest parsing params.r=rz Short description Args: name: description 1 priority (int): description 2 sender (str?): description 3 ratio (Optional[float], optional): description 4 r-nameN description 1r'priorityint description 2r)senderstr description 3r+ratioOptional[float] description 4z Short description Args: name: description 1 with multi-line text priority (int): description 2 "description 1 with multi-line textr rrrrbr3rar\s r test_paramsrz)**I y A % % % %   I y A % % % %  A  '6 1 1 1 1  A  ( 0 0 0  A  *o = = = ="... .  A  ': 5 5 5 5  A  (E 1 1 1 1  A  *o = = = ="... .  A  '8 3 3 3 3  A  (E 1 1 1 1  A  *o = = = =  A  *** *  A  '7 2 2 2 2  A  (,= = = = =  A  *o = = = =  A  *** *   I y A % % % %  A  '6 1 1 1 1  A  ( 0 0 0  A  *-      A  ': 5 5 5 5  A  (E 1 1 1 1  A  *o = = = = = =rctd}t|jdksJtd}t|jdksJ|jdjdksJ|jdjJ|jdjdksJ|jdjrJ|jdjd ksJ|jdjd ksJ|jdjd ksJ|jdjrJ|jd jd ksJ|jd jdksJ|jd jdksJ|jd jsJ|jdjdksJ|jdjdksJ|jdjdksJ|jdjsJtd}t|jd ksJ|jdjdksJ|jdjJ|jdjdksJ|jdjd ksJ|jdjd ksJ|jdjd ksJdS)zTest parsing attributes.r=rz Short description Attributes: name: description 1 priority (int): description 2 sender (str?): description 3 ratio (Optional[float], optional): description 4 r-rmNrnr'rorprqr)rrrsrtr+rurvrwz Short description Attributes: name: description 1 with multi-line text priority (int): description 2 rxryr\s rtest_attributesr}r{rctd}|jJ|jJt|jdksJtd}|jJ|jjJ|jjdksJ|jJt|jdksJ|jd|jksJtd}|jJ|jjJ|jjdksJ|jJt|jdksJ|jd|jksJtd }|jJ|jjd ksJ|jjdksJ|jJt|jdksJ|jd|jksJtd }|jJ|jjd ksJ|jjd ksJ|jJt|jdksJ|jd|jksJtd}|jJ|jjd ksJ|jjdksJ|jJt|jdksJ|jd|jksJtd}|jJ|jjd ksJ|jjdksJ|jJt|jdksJ|jd|jksJdS)zTest parsing returns.# Short description NrzL Short description Returns: description r3r'z[ Short description Returns: description with: a colon! zdescription with: a colon!zQ Short description Returns: int: description rpze Returns: Optional[Mapping[str, List[int]]]: A description: with a colon z!Optional[Mapping[str, List[int]]]zA description: with a colonzP Short description Yields: int: description z Short description Returns: int: description with much text even some spacing z-description with much text even some spacing)r returns many_returnsrrbr3r\s r test_returnsr s I   $ $ $  ! - - - y% & &! + + + + I   ( ( (   & . . .   (M 9 9 9 9  ! - - - y% & &! + + + +  !! $ (9 9 9 9 9 I   ( ( (   & . . .   (,H H H H H  ! - - - y% & &! + + + +  !! $ (9 9 9 9 9 I   ( ( (   &% / / / /   (M 9 9 9 9  ! - - - y% & &! + + + +  !! $ (9 9 9 9 9 I   ( ( (   &*M M M M M   (,I I I I I  ! - - - y% & &! + + + +  !! $ (9 9 9 9 9 I   ( ( (   &% / / / /   (M 9 9 9 9  ! - - - y% & &! + + + +  !! $ (9 9 9 9 9   I   ( ( (   &% / / / /   (:      ! - - - y% & &! + + + +  !! $ (9 9 9 9 9 9 9rc td}t|jdksJtd}t|jdksJ|jdjdksJ|jdjdksJdS)zTest parsing raises.rrzW Short description Raises: ValueError: description r' ValueErrorr3N)r rrfrbr3r\s r test_raisesrps I y A % % % % I y A % % % %  A  (L 8 8 8 8  A  *m ; ; ; ; ; ;rctd}t|jdksJ|jdjdksJ|jdjdksJdS)zTest parsing examples.z Short description Example: example: 1 Examples: long example more here r)rz example: 1r'zlong example more hereN)r rr#r3r\s r test_examplesrss   I y! " "a ' ' ' '  a , < < < <  a ,0K K K K K K Krctjt5tddddn #1swxYwYtjt5tdddddS#1swxYwYdS)zTest parsing broken meta.zArgs:NzArgs: herp derppytestrfrr rrtest_broken_metars z " " g z " "&& $%%%&&&&&&&&&&&&&&&&&&s6::A77A;>A;ctd}|jdjdksJ|jdjdksJ|jdjdksJ|jdjdksJdS) zTest parsing unknown meta.zShort desc Unknown 0: title0: content0 Args: arg0: desc0 arg1: desc1 Unknown1: title1: content1 Unknown2: title2: content2 rarg0desc0r'r(desc1N)r rrr3r\s rtest_unknown_metars I$  A  '6 1 1 1 1  A  *g 5 5 5 5  A  '6 1 1 1 1  A  *g 5 5 5 5 5 5rctjt5tdddddS#1swxYwYdS)zTest parsing broken arguments.zWThis is a test Args: param - poorly formatted Nrrrrtest_broken_argumentsrs z " "                         s 7;;ctd}t|jdksJ|jdjdgksJ|jdjdksJdS)z$Test parsing empty examples section.z]Short description Example: Raises: IOError: some error r'rr#r:N)r rr#r2r3r\s rtest_empty_examplerst I y! " "a ' ' ' '  a %* 5 5 5 5  a , 2 2 2 2 2 2r))r:r:)r;r:r<r>r?)rDrD)rFrD)rGz/Short description Long description Second line)rIrI)rJrI)rKrI)rLz.Short description Long description Second line)zQ Short description Meta: asd zShort description Meta: asd)zn Short description Long description Meta: asd z0Short description Long description Meta: asd)z Short description First line Second line Meta: asd z:Short description First line Second line Meta: asd)z Short description First line Second line Meta: asd z;Short description First line Second line Meta: asd)z Short description First line Second line Meta: asd z> f ~/CDDD       r)ra1Short description Args: name: description 1 priority (int): description 2 sender (str, optional): description 3 message (str, optional): description 4, defaults to 'hello' multiline (str, optional): long description 5, defaults to 'bye'cbtt|tj|ksJdS)zTest compose in expanded mode.rN)rr rEXPANDEDrs rtest_compose_expandedrs?H f ~/FGGG       r)r N)(__doc__typingTrdocstring_parser.commonrrdocstring_parser.googlerrrrr rrr4r9mark parametrizersrCboolrROptionalrXr]rdrkrzr}rrrrrrrrrrrrrrs_00 >>>>>>>> $ $ $ $ 9999,.K.K.K.Kb%5%5%5%5P  3#$  E;;;>>~          >>~ :LLLPPb $  $C $ 3 $ $ $ # $  $ $ $cPPb $ ????(AAAA:3333<4>4>4>4>n4>4>4>4>na:a:a:a:H<<<<*LLLL$&&&&66664     3333"~~~AAD......EAAD.  8scd98   B#C  Br