Add new test case

Author: cdce8p

tests/fixtures/comment_3.py

@@ -1,10 +1,7 @@
 """Comment will be changed. Only valid with '--force'."""
 import sys
-from typing import Any, List, Union
-
-# This is a comment
-from xyz import not_exist
-
+from typing import (  # this comment should not be deleted
+    Any, List, Union)
 
 var: Any = sys.version
 var2: List[int]

tests/fixtures/comment_3_forced.py

@@ -4,11 +4,6 @@
 import sys
 from typing import Any
 
-from xyz import not_exist
-
-# This is a comment
-
-
 var: Any = sys.version
 var2: list[int]
 

tests/fixtures/comment_3_no_change.py

@@ -1,10 +1,7 @@
 """Comment will be changed. Only valid with '--force'."""
 import sys
-from typing import Any, List, Union
-
-# This is a comment
-from xyz import not_exist
-
+from typing import (  # this comment should not be deleted
+    Any, List, Union)
 
 var: Any = sys.version
 var2: List[int]

tests/fixtures/comment_4.py

@@ -0,0 +1,15 @@
+"""Comment will be changed. Only valid with '--force'."""
+import sys
+from typing import Any, List, Union
+
+# This is a comment
+from xyz import not_exist
+
+var: Any = sys.version
+var2: List[int]
+
+
+def func(
+    arg: Union[int, str],
+) -> List[int]:
+    pass

tests/fixtures/comment_4_forced.py

@@ -0,0 +1,18 @@
+"""Comment will be changed. Only valid with '--force'."""
+from __future__ import annotations
+
+import sys
+from typing import Any
+
+from xyz import not_exist
+
+# This is a comment
+
+var: Any = sys.version
+var2: list[int]
+
+
+def func(
+    arg: int | str,
+) -> list[int]:
+    pass

tests/fixtures/comment_4_no_change.py

@@ -0,0 +1,15 @@
+"""Comment will be changed. Only valid with '--force'."""
+import sys
+from typing import Any, List, Union
+
+# This is a comment
+from xyz import not_exist
+
+var: Any = sys.version
+var2: List[int]
+
+
+def func(
+    arg: Union[int, str],
+) -> List[int]:
+    pass

tests/test_main.py

@@ -106,21 +106,31 @@ async def async_restore_fixtures(file_list: list[str]) -> AsyncGenerator[None, N
             None, 2,
             id="23_comment_3_no_change",
         ),
+        pytest.param(
+            'comment_4.py', 'comment_4_no_change.py',
+            None, 2,
+            id="24_comment_4_no_change",
+        ),
         pytest.param(
             'comment_1.py', 'comment_1_forced.py',
             ['--force'], 2,
-            id="24_comment_1_forced",
+            id="25_comment_1_forced",
             marks=pytest.mark.xfail(reason="Not implemented yet")
         ),
         pytest.param(
             'comment_2.py', 'comment_2_forced.py',
             ['--force'], 2,
-            id="25_comment_2_forced",
+            id="26_comment_2_forced",
         ),
         pytest.param(
             'comment_3.py', 'comment_3_forced.py',
             ['--force'], 2,
-            id="26_comment_3_forced",
+            id="27_comment_3_forced",
+        ),
+        pytest.param(
+            'comment_4.py', 'comment_4_forced.py',
+            ['--force'], 2,
+            id="28_comment_4_forced",
         ),
         pytest.param(
             'comment_no_issue_1.py', 'comment_no_issue_1_fixed.py',