gh-95591: [Enum] use _FlagTests base class (GH-96475)

(cherry picked from commit 9b58eaf98c)

Co-authored-by: Nikita Sobolev <mail@sobolevn.me>
This commit is contained in:
Miss Islington (bot) 2022-09-20 16:34:14 -07:00 committed by GitHub
parent 0e61d2b557
commit fd118a0850
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -764,7 +764,7 @@ class _FlagTests:
def test_default_missing_with_wrong_type_value(self):
with self.assertRaisesRegex(
ValueError,
"'RED' is not a valid TestFlag.Color",
"'RED' is not a valid ",
) as ctx:
self.MainEnum('RED')
self.assertIs(ctx.exception.__context__, None)
@ -773,7 +773,7 @@ class TestPlainEnum(_EnumTests, _PlainOutputTests, unittest.TestCase):
enum_type = Enum
class TestPlainFlag(_EnumTests, _PlainOutputTests, unittest.TestCase):
class TestPlainFlag(_EnumTests, _PlainOutputTests, _FlagTests, unittest.TestCase):
enum_type = Flag
@ -785,7 +785,7 @@ class TestStrEnum(_EnumTests, _MinimalOutputTests, unittest.TestCase):
enum_type = StrEnum
class TestIntFlag(_EnumTests, _MinimalOutputTests, unittest.TestCase):
class TestIntFlag(_EnumTests, _MinimalOutputTests, _FlagTests, unittest.TestCase):
enum_type = IntFlag
@ -797,7 +797,7 @@ class TestMixedStr(_EnumTests, _MixedOutputTests, unittest.TestCase):
class enum_type(str, Enum): pass
class TestMixedIntFlag(_EnumTests, _MixedOutputTests, unittest.TestCase):
class TestMixedIntFlag(_EnumTests, _MixedOutputTests, _FlagTests, unittest.TestCase):
class enum_type(int, Flag): pass