Add support for pyi files (#210)

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

diff --git a/tests/comments4.py b/tests/comments4.py
index 241b6cee72e11e8e8b8b6d937a2f5aaea10b0a68..95299900af7df6ffd49fabe6074722665e52a8e9 100644 (file)
--- a/tests/comments4.py
+++ b/tests/comments4.py
@@ -63,14 +63,26 @@ def foo(list_a, list_b):
     results = (
         User.query.filter(User.foo == "bar").filter(  # Because foo.
             db.or_(User.field_a.astext.in_(list_a), User.field_b.astext.in_(list_b))
-        ).filter(
-            User.xyz.is_(None)
-        )
+        ).filter(User.xyz.is_(None))
         # Another comment about the filtering on is_quux goes here.
         .filter(db.not_(User.is_pending.astext.cast(db.Boolean).is_(True))).order_by(
             User.created_at.desc()
-        ).with_for_update(
-            key_share=True
-        ).all()
+        ).with_for_update(key_share=True).all()
     )
     return results
+
+
+def foo2(list_a, list_b):
+    # Standalone comment reasonably placed.
+    return User.query.filter(User.foo == "bar").filter(
+        db.or_(User.field_a.astext.in_(list_a), User.field_b.astext.in_(list_b))
+    ).filter(User.xyz.is_(None))
+
+
+def foo3(list_a, list_b):
+    return (
+        # Standlone comment but weirdly placed.
+        User.query.filter(User.foo == "bar").filter(
+            db.or_(User.field_a.astext.in_(list_a), User.field_b.astext.in_(list_b))
+        ).filter(User.xyz.is_(None))
+    )