mirrors
/
cpython
mirror of https://github.com/python/cpython.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

gh-121671: Increase test coverage of `ast.get_docstring` (GH-121674) 

Increase test coverage for `ast.get_docstring`
This commit is contained in:
Tomas R 2024-07-13 14:59:15 +02:00 committed by GitHub
parent fc21781175
commit 0a26aa5007
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
Lib/test/test_ast.py
View File

@ -1821,6 +1821,12 @@ def test_get_docstring(self):
node = ast.parse('async def foo():\n """spam\n ham"""') node = ast.parse('async def foo():\n """spam\n ham"""')
self.assertEqual(ast.get_docstring(node.body[0]), 'spam\nham') self.assertEqual(ast.get_docstring(node.body[0]), 'spam\nham')
node = ast.parse('async def foo():\n """spam\n ham"""')
self.assertEqual(ast.get_docstring(node.body[0], clean=False), 'spam\n ham')
node = ast.parse('x')
self.assertRaises(TypeError, ast.get_docstring, node.body[0])
def test_get_docstring_none(self): def test_get_docstring_none(self):
self.assertIsNone(ast.get_docstring(ast.parse(''))) self.assertIsNone(ast.get_docstring(ast.parse('')))
node = ast.parse('x = "not docstring"') node = ast.parse('x = "not docstring"')
@ -1845,6 +1851,9 @@ def test_get_docstring_none(self):
node = ast.parse('async def foo():\n x = "not docstring"') node = ast.parse('async def foo():\n x = "not docstring"')
self.assertIsNone(ast.get_docstring(node.body[0])) self.assertIsNone(ast.get_docstring(node.body[0]))
node = ast.parse('async def foo():\n 42')
self.assertIsNone(ast.get_docstring(node.body[0]))
def test_multi_line_docstring_col_offset_and_lineno_issue16806(self): def test_multi_line_docstring_col_offset_and_lineno_issue16806(self):
node = ast.parse( node = ast.parse(
'"""line one\nline two"""\n\n' '"""line one\nline two"""\n\n'