Expand tests

astral-sh · Aug 18, 2023 · 68fe3f1 · 68fe3f1
1 parent c2098c6
commit 68fe3f1
Show file tree

Hide file tree

Showing 2 changed files with 98 additions and 2 deletions.
diff --git a/crates/ruff_python_formatter/resources/test/fixtures/ruff/statement/ellipsis.pyi b/crates/ruff_python_formatter/resources/test/fixtures/ruff/statement/ellipsis.pyi
@@ -3,15 +3,48 @@
 while True:
     ...
 
+if True:
+    # comment
+    ...
+
+with True:
+    ...  # comment
+
+for i in []:
+    ...
+
+for i in []:
+    # comment
+    ...
+
 for i in []:
+    ... # comment
+
+if True:
+    ...
+
+if True:
+    # comment
     ...
 
 if True:
+    ... # comment
+
+with True:
     ...
 
 with True:
+    # comment
     ...
 
+with True:
+    ... # comment
+
 match x:
     case 1:
-        ...
+        ...
+    case 2:
+        # comment
+        ...
+    case 3:
+        ... # comment
diff --git a/crates/ruff_python_formatter/tests/snapshots/format@statement__ellipsis.pyi.snap b/crates/ruff_python_formatter/tests/snapshots/format@statement__ellipsis.pyi.snap
@@ -9,34 +9,97 @@ input_file: crates/ruff_python_formatter/resources/test/fixtures/ruff/statement/
 while True:
     ...
 
+if True:
+    # comment
+    ...
+
+with True:
+    ...  # comment
+
 for i in []:
     ...
 
+for i in []:
+    # comment
+    ...
+
+for i in []:
+    ... # comment
+
 if True:
     ...
 
+if True:
+    # comment
+    ...
+
+if True:
+    ... # comment
+
+with True:
+    ...
+
 with True:
+    # comment
     ...
 
+with True:
+    ... # comment
+
 match x:
     case 1:
-        ...```
+        ...
+    case 2:
+        # comment
+        ...
+    case 3:
+        ... # comment
+```
 
 ## Output
 ```py
 """Compound statements with no body should be written on one line."""
 
 while True: ...
 
+if True:
+    # comment
+    ...
+
+with True: ...  # comment
+
 for i in []: ...
 
+for i in []:
+    # comment
+    ...
+
+for i in []: ...  # comment
+
 if True: ...
 
+if True:
+    # comment
+    ...
+
+if True: ...  # comment
+
 with True: ...
 
+with True:
+    # comment
+    ...
+
+with True: ...  # comment
+
 match x:
     case "NOT_YET_IMPLEMENTED_PatternMatchValue":
         ...
+    case "NOT_YET_IMPLEMENTED_PatternMatchValue":
+        # comment
+        ...
+    case "NOT_YET_IMPLEMENTED_PatternMatchValue":
+        ...  # comment
 ```