diff options
| author | Yurii Karabas <1998uriyyo@gmail.com> | 2021-07-20 14:48:05 (GMT) |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-07-20 14:48:05 (GMT) |
| commit | 4868b94c6089d457673b1ba5b5b64c2f38c435af (patch) | |
| tree | edb0ae78d3b36cc40d6d73b38d27f66ebc471add /Lib/test/test_typing.py | |
| parent | 965dd76e9060e27e2253ba8c8d21a142b178720d (diff) | |
| download | cpython-4868b94c6089d457673b1ba5b5b64c2f38c435af.zip cpython-4868b94c6089d457673b1ba5b5b64c2f38c435af.tar.gz cpython-4868b94c6089d457673b1ba5b5b64c2f38c435af.tar.bz2 | |
bpo-44353: Add test to cover __or__ of two NewType (#27259)
Diffstat (limited to 'Lib/test/test_typing.py')
| -rw-r--r-- | Lib/test/test_typing.py | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/Lib/test/test_typing.py b/Lib/test/test_typing.py index 169b92b..ba51b9c 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) |