From 4868b94c6089d457673b1ba5b5b64c2f38c435af Mon Sep 17 00:00:00 2001 From: Yurii Karabas <1998uriyyo@gmail.com> Date: Tue, 20 Jul 2021 17:48:05 +0300 Subject: [PATCH] bpo-44353: Add test to cover __or__ of two NewType (#27259) --- Lib/test/test_typing.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/Lib/test/test_typing.py b/Lib/test/test_typing.py index 169b92b6513..ba51b9c3e06 100644 --- a/Lib/test/test_typing.py +++ b/Lib/test/test_typing.py @@ -3693,12 +3693,15 @@ class D(UserName): 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)