Add more tests for function doc

palaviv · palaviv · commit c76b31866d2f · 2019-03-31T18:52:27.000+03:00
diff --git a/tests/snippets/function.py b/tests/snippets/function.py
@@ -5,8 +5,35 @@ def foo():
 assert foo() == 42
 assert foo.__doc__ == "test"
 
-
 def my_func(a,):
     return a+2
 
 assert my_func(2) == 4
+
+
+def f1():
+
+    """test1"""
+    pass
+
+assert f1.__doc__ == "test1"
+
+def f2():
+    '''test2'''
+    pass
+
+assert f2.__doc__ == "test2"
+
+def f3():
+    """
+    test3
+    """
+    pass
+
+assert f3.__doc__ == "\n    test3\n    "
+
+def f4():
+    "test4"
+    pass
+
+assert f4.__doc__ == "test4"
