From: Yurii Karabas <1998uriyyo@gmail.com>
Date: Tue, 20 Jul 2021 14:48:05 +0000 (+0300)
Subject: bpo-44353: Add test to cover __or__ of two NewType (#27259)
X-Git-Tag: v3.11.0a1~621
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=4868b94c6089d457673b1ba5b5b64c2f38c435af;p=thirdparty%2FPython%2Fcpython.git

bpo-44353: Add test to cover __or__ of two NewType (#27259)
---

diff --git a/Lib/test/test_typing.py b/Lib/test/test_typing.py
index 169b92b65137..ba51b9c3e066 100644
--- a/Lib/test/test_typing.py
+++ b/Lib/test/test_typing.py
@@ -3693,12 +3693,15 @@ class NewTypeTests(BaseTestCase):
 
     def test_or(self):
         UserId = NewType('UserId', int)
+        UserName = NewType('UserName', str)
 
-        self.assertEqual(UserId | int, Union[UserId, int])
-        self.assertEqual(int | UserId, Union[int, UserId])
+        for cls in (int, UserName):
+            with self.subTest(cls=cls):
+                self.assertEqual(UserId | cls, Union[UserId, cls])
+                self.assertEqual(cls | UserId, Union[cls, UserId])
 
-        self.assertEqual(get_args(UserId | int), (UserId, int))
-        self.assertEqual(get_args(int | UserId), (int, UserId))
+                self.assertEqual(get_args(UserId | cls), (UserId, cls))
+                self.assertEqual(get_args(cls | UserId), (cls, UserId))
 
     def test_special_attrs(self):
         UserId = NewType('UserId', int)