Skip to content

Commit fd118a0

Browse files
gh-95591: [Enum] use _FlagTests base class (GH-96475)
(cherry picked from commit 9b58eaf) Co-authored-by: Nikita Sobolev <[email protected]>
1 parent 0e61d2b commit fd118a0

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

Lib/test/test_enum.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -764,7 +764,7 @@ class _FlagTests:
764764
def test_default_missing_with_wrong_type_value(self):
765765
with self.assertRaisesRegex(
766766
ValueError,
767-
"'RED' is not a valid TestFlag.Color",
767+
"'RED' is not a valid ",
768768
) as ctx:
769769
self.MainEnum('RED')
770770
self.assertIs(ctx.exception.__context__, None)
@@ -773,7 +773,7 @@ class TestPlainEnum(_EnumTests, _PlainOutputTests, unittest.TestCase):
773773
enum_type = Enum
774774

775775

776-
class TestPlainFlag(_EnumTests, _PlainOutputTests, unittest.TestCase):
776+
class TestPlainFlag(_EnumTests, _PlainOutputTests, _FlagTests, unittest.TestCase):
777777
enum_type = Flag
778778

779779

@@ -785,7 +785,7 @@ class TestStrEnum(_EnumTests, _MinimalOutputTests, unittest.TestCase):
785785
enum_type = StrEnum
786786

787787

788-
class TestIntFlag(_EnumTests, _MinimalOutputTests, unittest.TestCase):
788+
class TestIntFlag(_EnumTests, _MinimalOutputTests, _FlagTests, unittest.TestCase):
789789
enum_type = IntFlag
790790

791791

@@ -797,7 +797,7 @@ class TestMixedStr(_EnumTests, _MixedOutputTests, unittest.TestCase):
797797
class enum_type(str, Enum): pass
798798

799799

800-
class TestMixedIntFlag(_EnumTests, _MixedOutputTests, unittest.TestCase):
800+
class TestMixedIntFlag(_EnumTests, _MixedOutputTests, _FlagTests, unittest.TestCase):
801801
class enum_type(int, Flag): pass
802802

803803

0 commit comments

Comments
 (0)