Format subscriptions in a PEP-8 compliant way (#178)

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

diff --git a/tests/function.py b/tests/function.py
index 007cc98ace10238fbf44d93832f944817c1f4a57..4ec90571298f60abc862df0ee68c57b52a3be7b7 100644 (file)
--- a/tests/function.py
+++ b/tests/function.py
@@ -33,7 +33,7 @@ def spaces(a=1, b=(), c=[], d={}, e=True, f=-1, g=1 if False else 2, h="", i=r''
  assert task._cancel_stack[:len(old_stack)] == old_stack
 def spaces_types(a: int = 1, b: tuple = (), c: list = [], d: dict = {}, e: bool = True, f: int = -1, g: int = 1 if False else 2, h: str = "", i: str = r''): ...
 def spaces2(result= _core.Value(None)):
  assert task._cancel_stack[:len(old_stack)] == old_stack
 def spaces_types(a: int = 1, b: tuple = (), c: list = [], d: dict = {}, e: bool = True, f: int = -1, g: int = 1 if False else 2, h: str = "", i: str = r''): ...
 def spaces2(result= _core.Value(None)):

- ...
+ assert fut is self._read_fut, (fut, self._read_fut)

     # EMPTY LINE WITH WHITESPACE (this comment will be removed)
 def example(session):
     result = session.query(models.Customer.id).filter(
     # EMPTY LINE WITH WHITESPACE (this comment will be removed)
 def example(session):
     result = session.query(models.Customer.id).filter(


@@ -47,6 +47,12 @@ def long_lines():
         typedargslist.extend(
             gen_annotated_params(ast_args.kwonlyargs, ast_args.kw_defaults, parameters, implicit_default=True)
         )
         typedargslist.extend(
             gen_annotated_params(ast_args.kwonlyargs, ast_args.kw_defaults, parameters, implicit_default=True)
         )

+        typedargslist.extend(
+            gen_annotated_params(
+                ast_args.kwonlyargs, ast_args.kw_defaults, parameters, implicit_default=True,
+                # trailing standalone comment
+            )
+        )

     _type_comment_re = re.compile(
         r"""
         ^
     _type_comment_re = re.compile(
         r"""
         ^


@@ -68,6 +74,13 @@ 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),
+}

 
 # output
 
 
 # output
 


@@ -89,13 +102,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
 


@@ -122,7 +133,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(


@@ -140,7 +151,7 @@ def spaces_types(
 
 
 def spaces2(result=_core.Value(None)):
 
 
 def spaces2(result=_core.Value(None)):

-    ...
+    assert fut is self._read_fut, (fut, self._read_fut)

 
 
 def example(session):
 
 
 def example(session):


@@ -161,6 +172,15 @@ def long_lines():
                 implicit_default=True,
             )
         )
                 implicit_default=True,
             )
         )

+        typedargslist.extend(
+            gen_annotated_params(
+                ast_args.kwonlyargs,
+                ast_args.kw_defaults,
+                parameters,
+                implicit_default=True,
+                # trailing standalone comment
+            )
+        )

     _type_comment_re = re.compile(
         r"""
         ^
     _type_comment_re = re.compile(
         r"""
         ^


@@ -183,3 +203,12 @@ 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),
+    }