Support sticky standalone comments (comments preceding defs, classes, and decorators)

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

diff --git a/tests/composition.py b/tests/composition.py
index 7b462ac717c6b8b794ead7c14406e88a8195c07f..287888d984e086251610070df4e2316cac7dadf7 100644 (file)
--- a/tests/composition.py
+++ b/tests/composition.py
@@ -3,19 +3,32 @@ class C:
     def test(self) -> None:
         with patch("black.out", print):
             self.assertEqual(
-                unstyle(str(report)), '1 file reformatted, 1 file failed to reformat.'
+                unstyle(str(report)), "1 file reformatted, 1 file failed to reformat."
             )
             self.assertEqual(
                 unstyle(str(report)),
-                '1 file reformatted, 1 file left unchanged, 1 file failed to reformat.',
+                "1 file reformatted, 1 file left unchanged, 1 file failed to reformat.",
             )
             self.assertEqual(
                 unstyle(str(report)),
-                '2 files reformatted, 1 file left unchanged, '
-                '1 file failed to reformat.',
+                "2 files reformatted, 1 file left unchanged, "
+                "1 file failed to reformat.",
             )
             self.assertEqual(
                 unstyle(str(report)),
-                '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)