Automatic management of parentheses in `elif`, too

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

diff --git a/tests/function.py b/tests/function.py
index 6cf6dbc9a1b4d16da5874ee3fbd0e5b870c2771c..a18121219f58a4da1dc5533e9693467378d83521 100644 (file)
--- a/tests/function.py
+++ b/tests/function.py
@@ -74,6 +74,22 @@ def long_lines():
         $
         """, re.MULTILINE | re.VERBOSE
     )
         $
         """, re.MULTILINE | re.VERBOSE
     )

+def trailing_comma():
+    mapping = {
+    A: 0.25 * (10.0 / 12),
+    B: 0.1 * (10.0 / 12),
+    C: 0.1 * (10.0 / 12),
+    D: 0.1 * (10.0 / 12),
+}
+def f(
+  a,
+  **kwargs,
+) -> A:
+    return A(
+        very_long_argument_name1=very_long_value_for_the_argument,
+        very_long_argument_name2=very_long_value_for_the_argument,
+        **kwargs,
+    )

 
 # output
 
 
 # output
 


@@ -95,13 +111,11 @@ def func_no_args():
     c
     if True:
         raise RuntimeError
     c
     if True:
         raise RuntimeError

-

     if False:
         ...
     for i in range(10):
         print(i)
         continue
     if False:
         ...
     for i in range(10):
         print(i)
         continue

-

     exec("new-style exec", {}, {})
     return None
 
     exec("new-style exec", {}, {})
     return None
 


@@ -128,7 +142,7 @@ def function_signature_stress_test(
 
 def spaces(a=1, b=(), c=[], d={}, e=True, f=-1, g=1 if False else 2, h="", i=r""):
     offset = attr.ib(default=attr.Factory(lambda: _r.uniform(10000, 200000)))
 
 def spaces(a=1, b=(), c=[], d={}, e=True, f=-1, g=1 if False else 2, h="", i=r""):
     offset = attr.ib(default=attr.Factory(lambda: _r.uniform(10000, 200000)))

-    assert task._cancel_stack[:len(old_stack)] == old_stack
+    assert task._cancel_stack[: len(old_stack)] == old_stack

 
 
 def spaces_types(
 
 
 def spaces_types(


@@ -198,3 +212,20 @@ def long_lines():
         """,
         re.MULTILINE | re.VERBOSE,
     )
         """,
         re.MULTILINE | re.VERBOSE,
     )

+
+
+def trailing_comma():
+    mapping = {
+        A: 0.25 * (10.0 / 12),
+        B: 0.1 * (10.0 / 12),
+        C: 0.1 * (10.0 / 12),
+        D: 0.1 * (10.0 / 12),
+    }
+
+
+def f(a, **kwargs) -> A:
+    return A(
+        very_long_argument_name1=very_long_value_for_the_argument,
+        very_long_argument_name2=very_long_value_for_the_argument,
+        **kwargs,
+    )