aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/Lib/test/test_enum.py
diff options
context:
space:
mode:
Diffstat (limited to 'Lib/test/test_enum.py')
-rw-r--r--Lib/test/test_enum.py16
1 files changed, 16 insertions, 0 deletions
diff --git a/Lib/test/test_enum.py b/Lib/test/test_enum.py
index 61060f3dc29..cf3e042de1a 100644
--- a/Lib/test/test_enum.py
+++ b/Lib/test/test_enum.py
@@ -1048,6 +1048,22 @@ class TestPlainEnumFunction(_EnumTests, _PlainOutputTests, unittest.TestCase):
class TestPlainFlagClass(_EnumTests, _PlainOutputTests, _FlagTests, unittest.TestCase):
enum_type = Flag
+ def test_none_member(self):
+ class FlagWithNoneMember(Flag):
+ A = 1
+ E = None
+
+ self.assertEqual(FlagWithNoneMember.A.value, 1)
+ self.assertIs(FlagWithNoneMember.E.value, None)
+ with self.assertRaisesRegex(TypeError, r"'FlagWithNoneMember.E' cannot be combined with other flags with |"):
+ FlagWithNoneMember.A | FlagWithNoneMember.E
+ with self.assertRaisesRegex(TypeError, r"'FlagWithNoneMember.E' cannot be combined with other flags with &"):
+ FlagWithNoneMember.E & FlagWithNoneMember.A
+ with self.assertRaisesRegex(TypeError, r"'FlagWithNoneMember.E' cannot be combined with other flags with \^"):
+ FlagWithNoneMember.A ^ FlagWithNoneMember.E
+ with self.assertRaisesRegex(TypeError, r"'FlagWithNoneMember.E' cannot be inverted"):
+ ~FlagWithNoneMember.E
+
class TestPlainFlagFunction(_EnumTests, _PlainOutputTests, _FlagTests, unittest.TestCase):
enum_type = Flag