Add support for pyi files (#210)

[etc/vim.git] / tests / function.py

diff --git a/tests/function.py b/tests/function.py
index 4ec90571298f60abc862df0ee68c57b52a3be7b7..4cfc945c6b2af62f19f61ed4244a87b37328c89a 100644 (file)
--- a/tests/function.py
+++ b/tests/function.py
@@ -81,6 +81,18 @@ def trailing_comma():
     C: 0.1 * (10.0 / 12),
     D: 0.1 * (10.0 / 12),
 }
+def f(
+  a,
+  **kwargs,
+) -> A:
+    return (
+        yield from A(
+            very_long_argument_name1=very_long_value_for_the_argument,
+            very_long_argument_name2=very_long_value_for_the_argument,
+            **kwargs,
+        )
+    )
+def __await__(): return (yield)
 
 # output
 
@@ -157,9 +169,7 @@ def spaces2(result=_core.Value(None)):
 def example(session):
     result = session.query(models.Customer.id).filter(
         models.Customer.account_id == account_id, models.Customer.email == email_address
-    ).order_by(
-        models.Customer.id.asc()
-    ).all()
+    ).order_by(models.Customer.id.asc()).all()
 
 
 def long_lines():
@@ -212,3 +222,17 @@ def trailing_comma():
         C: 0.1 * (10.0 / 12),
         D: 0.1 * (10.0 / 12),
     }
+
+
+def f(a, **kwargs) -> A:
+    return (
+        yield from A(
+            very_long_argument_name1=very_long_value_for_the_argument,
+            very_long_argument_name2=very_long_value_for_the_argument,
+            **kwargs,
+        )
+    )
+
+
+def __await__():
+    return (yield)