Skip to content

Commit 9b58eaf

Browse files
authored
gh-95591: [Enum] use _FlagTests base class (GH-96475)
1 parent a3e2f05 commit 9b58eaf

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
@@ -777,7 +777,7 @@ class _FlagTests:
777777
def test_default_missing_with_wrong_type_value(self):
778778
with self.assertRaisesRegex(
779779
ValueError,
780-
"'RED' is not a valid TestFlag.Color",
780+
"'RED' is not a valid ",
781781
) as ctx:
782782
self.MainEnum('RED')
783783
self.assertIs(ctx.exception.__context__, None)
@@ -786,7 +786,7 @@ class TestPlainEnum(_EnumTests, _PlainOutputTests, unittest.TestCase):
786786
enum_type = Enum
787787

788788

789-
class TestPlainFlag(_EnumTests, _PlainOutputTests, unittest.TestCase):
789+
class TestPlainFlag(_EnumTests, _PlainOutputTests, _FlagTests, unittest.TestCase):
790790
enum_type = Flag
791791

792792

@@ -798,7 +798,7 @@ class TestStrEnum(_EnumTests, _MinimalOutputTests, unittest.TestCase):
798798
enum_type = StrEnum
799799

800800

801-
class TestIntFlag(_EnumTests, _MinimalOutputTests, unittest.TestCase):
801+
class TestIntFlag(_EnumTests, _MinimalOutputTests, _FlagTests, unittest.TestCase):
802802
enum_type = IntFlag
803803

804804

@@ -810,7 +810,7 @@ class TestMixedStr(_EnumTests, _MixedOutputTests, unittest.TestCase):
810810
class enum_type(str, Enum): pass
811811

812812

813-
class TestMixedIntFlag(_EnumTests, _MixedOutputTests, unittest.TestCase):
813+
class TestMixedIntFlag(_EnumTests, _MixedOutputTests, _FlagTests, unittest.TestCase):
814814
class enum_type(int, Flag): pass
815815

816816

0 commit comments

Comments
 (0)