Fix repr for union

yonmilk · yonmilk · commit cc5f88f301c0 · 2022-10-29T03:24:03.000+09:00
diff --git a/Lib/test/test_types.py b/Lib/test/test_types.py
@@ -929,8 +929,6 @@ def test_or_type_operator_with_SpecialForm(self):
         assert typing.Optional[int] | str == typing.Union[int, str, None]
         assert typing.Union[int, bool] | str == typing.Union[int, bool, str]
 
-    # TODO: RUSTPYTHON
-    @unittest.expectedFailure
     def test_or_type_repr(self):
         assert repr(int | str) == "int | str"
         assert repr((int | str) | list) == "int | str | list"
diff --git a/vm/src/builtins/union.rs b/vm/src/builtins/union.rs
@@ -64,7 +64,9 @@ impl PyUnion {
                     .and_then(|o| o.downcast_ref::<PyStr>().map(|m| m.as_str().to_string())),
             ) {
                 (None, _) | (_, None) => Ok(obj.repr(vm)?.as_str().to_string()),
-                (Some(qualname), Some(module)) => Ok(if module == "builtins" {
+                (Some(qualname), Some(module)) => Ok(if qualname.eq("NoneType") {
+                    "None".to_string()
+                } else if module == "builtins" {
                     qualname
                 } else {
                     format!("{}.{}", module, qualname)
