Add support for | unions from dmypy suggest argument type annotations

Author: CoolCat467

src/idlemypyextension/annotate.py

@@ -176,7 +176,7 @@ def tokenize(txt: str) -> list[Token]:
             return tokens
         if txt[0] in {" ", "\n"}:
             txt = txt[1:]
-        elif txt[0] in "()[],*":
+        elif txt[0] in "()[],*|":
             tokens.append(Separator(txt[0]))
             txt = txt[1:]
         elif txt[:2] == "->":
@@ -983,7 +983,7 @@ def get_line(line: int) -> str:
                     return line_text
             raise EOFError
 
-    print(f"{get_annotation(annotation, get_line)!r}")
+    print(f"{get_annotation(annotation, get_line)[0]}")
 
 
 if __name__ == "__main__":  # pragma: nocover

tests/test_annotate.py

@@ -175,6 +175,14 @@ def test_list_or(items: Collection[str], result: str) -> None:
                 annotate.End(),
             ],
         ),
+        (
+            "int | float",
+            [
+                annotate.DottedName("int"),
+                annotate.Separator("|"),
+                annotate.DottedName("float"),
+            ],
+        ),
     ],
 )
 def test_tokenize(text: str, tokens: list[annotate.Token]) -> None: