Fix an embarrassing UnboundLocalError

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

diff --git a/tests/composition.py b/tests/composition.py
index fb27b3e55304914e2cb7f3019ed43de7de211280..287888d984e086251610070df4e2316cac7dadf7 100644 (file)
--- a/tests/composition.py
+++ b/tests/composition.py
@@ -19,3 +19,16 @@ class C:
                 "2 files reformatted, 2 files left unchanged, "
                 "2 files failed to reformat.",
             )
                 "2 files reformatted, 2 files left unchanged, "
                 "2 files failed to reformat.",
             )

+            for i in (a,):
+                if (
+                    # Rule 1
+                    i % 2 == 0
+                    # Rule 2
+                    and i % 3 == 0
+                ):
+                    while (
+                        # Just a comment
+                        call()
+                        # Another
+                    ):
+                        print(i)