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

This commit is contained in:
Yurii Karabas 2021-07-20 17:48:05 +03:00 committed by GitHub
parent 965dd76e90
commit 4868b94c60
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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)