diff options
Diffstat (limited to 'Lib')
| -rw-r--r-- | Lib/compiler/transformer.py | 1 | ||||
| -rw-r--r-- | Lib/test/test_compiler.py | 8 |
2 files changed, 9 insertions, 0 deletions
diff --git a/Lib/compiler/transformer.py b/Lib/compiler/transformer.py index 53d30f0..a16dc55 100644 --- a/Lib/compiler/transformer.py +++ b/Lib/compiler/transformer.py @@ -1382,6 +1382,7 @@ _doc_nodes = [ symbol.testlist, symbol.testlist_safe, symbol.test, + symbol.or_test, symbol.and_test, symbol.not_test, symbol.comparison, diff --git a/Lib/test/test_compiler.py b/Lib/test/test_compiler.py index 17f181e..929a12b 100644 --- a/Lib/test/test_compiler.py +++ b/Lib/test/test_compiler.py @@ -68,6 +68,14 @@ class CompilerTest(unittest.TestCase): def testDefaultArgs(self): self.assertRaises(SyntaxError, compiler.parse, "def foo(a=1, b): pass") + def testDocstrings(self): + c = compiler.compile('"doc"', '<string>', 'exec') + self.assert_('__doc__' in c.co_names) + c = compiler.compile('def f():\n "doc"', '<string>', 'exec') + g = {} + exec c in g + self.assertEquals(g['f'].__doc__, "doc") + def testLineNo(self): # Test that all nodes except Module have a correct lineno attribute. filename = __file__ |