o DDi@sPddlmZddlmZddlZddlmZddlm Z m Z Gddde Z dS))dedent)OptionalN) ensure_type) data_providerUnitTestc@s|eZdZeddedeeddfddZeddedeeddfd d Zed dedeeddfd d ZdddZ dS) DocstringTest))N)z""r)z# comment is not docstringN)z] # comment """docstring in triple quotes.""" zdocstring in triple quotes.)z"docstring in single quotes."zdocstring in single quotes.)zg # comment """docstring in """ "concatenated strings." z"docstring in concatenated strings.code docstringreturnNcCs|tt||dSN assertEqualcst parse_moduler get_docstringselfr r r[/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/_nodes/tests/test_docstring.pytest_module_docstringsz#DocstringTest.test_module_docstring))zO def f(): # comment" pass N)zdef f():"docstring"r )z def f(): """ This function has no input and always returns None. """ z3This function has no input and always returns None.)z def fn(): # comment 1 # comment 2 "docstring" r )zM def fn(): ("docstring") r cC&|ttt|tj|dSr )rrrparse_statementr FunctionDefrrrrrtest_function_docstring,s)z%DocstringTest.test_function_docstring))zO class C: # comment" pass N)zclass C(Base):"docstring"r )z class C(Base): # a comment """ This class has a multi- line docstring. """ z'This class has a multi- line docstring.)zm class C(A, B): # comment 1 # comment 2 "docstring" r cCrr )rrrrrClassDefrrrrrtest_class_docstring\s!z"DocstringTest.test_class_docstringcCs@d}|tt|d|tt|jddddS)Nz """ A docstring with indentation one first line and the second line. """ z@A docstring with indentation one first line and the second line.F)cleanzF A docstring with indentation one first line and the second line. r )rr rrrtest_clean_docstringsz"DocstringTest.test_clean_docstring)r N) __name__ __module__ __qualname__rstrrrrrrrrrrrs( r) textwraprtypingrlibcstrlibcst.helpersrlibcst.testing.utilsrrrrrrrs