statements should care for its own docstrings, not the parser

jedi/parsing.py

@@ -413,6 +413,7 @@ class Parser(object):
             with common.ignored(IndexError, AttributeError):
                 # ... then set it as a docstring
                 self._scope.statements[-1].add_docstr(first_tok[1])
+                return None, tok
 
         if tok in always_break + not_first_break:
             self._gen.push_last_back()

jedi/parsing_representation.py

@@ -800,6 +800,8 @@ class Statement(Simple):
 
         code = ''.join(assemble(*a) for a in self.assignment_details)
         code += assemble(self.get_commands())
+        if self.docstr:
+            code += '\n"""%s"""' % self.docstr
 
         if new_line:
             return code + '\n'

test/test_parsing_representation.py

@@ -7,7 +7,7 @@ def test_get_code():
 class SomeClass(object, mixin):
     def __init__(self):
         self.xy = 3.0
-        'statement docstr'
+        """statement docstr"""
     def some_method(self):
         return 1
     def yield_method(self):