Fix compiled docstrings for stubs

2018-09-04 10:08:09 +02:00
parent 5d24bc7625
commit e086c433ff
3 changed files with 36 additions and 12 deletions
--- a/test/test_evaluate/test_compiled.py
+++ b/test/test_evaluate/test_compiled.py
@@ -77,7 +77,9 @@ def test_method_completion(Script, environment):
 def test_time_docstring(Script):
    import time
    comp, = Script('import time\ntime.sleep').completions()
-    assert comp.docstring() == time.sleep.__doc__
+    assert comp.docstring(raw=True) == time.sleep.__doc__
+    expected = 'sleep(secs: float) -> None\n\n' + time.sleep.__doc__
+    assert comp.docstring() == expected


 def test_dict_values(Script):