Skip to content

Commit a4fe025

Browse files
committed
gh-71339: Use new assertion methods in test_enum
1 parent 7239da7 commit a4fe025

File tree

1 file changed

+22
-22
lines changed

1 file changed

+22
-22
lines changed

Lib/test/test_enum.py

Lines changed: 22 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -433,9 +433,9 @@ class Season(self.enum_type):
433433
def spam(cls):
434434
pass
435435
#
436-
self.assertTrue(hasattr(Season, 'spam'))
436+
self.assertHasAttr(Season, 'spam')
437437
del Season.spam
438-
self.assertFalse(hasattr(Season, 'spam'))
438+
self.assertNotHasAttr(Season, 'spam')
439439
#
440440
with self.assertRaises(AttributeError):
441441
del Season.SPRING
@@ -1415,7 +1415,7 @@ class Outer(Enum):
14151415
class Inner(Enum):
14161416
foo = 10
14171417
bar = 11
1418-
self.assertTrue(isinstance(Outer.Inner, Outer))
1418+
self.assertIsInstance(Outer.Inner, Outer)
14191419
self.assertEqual(Outer.a.value, 1)
14201420
self.assertEqual(Outer.Inner.value.foo.value, 10)
14211421
self.assertEqual(
@@ -1439,7 +1439,7 @@ class Outer(Enum):
14391439
class Inner(Enum):
14401440
foo = 10
14411441
bar = 11
1442-
self.assertTrue(isinstance(Outer.Inner, type))
1442+
self.assertIsInstance(Outer.Inner, type)
14431443
self.assertEqual(Outer.a.value, 1)
14441444
self.assertEqual(Outer.Inner.foo.value, 10)
14451445
self.assertEqual(
@@ -1459,7 +1459,7 @@ class Outer(Enum):
14591459
class Inner(Enum):
14601460
foo = 10
14611461
bar = 11
1462-
self.assertTrue(isinstance(Outer.Inner, type))
1462+
self.assertIsInstance(Outer.Inner, type)
14631463
self.assertEqual(Outer.a.value, 1)
14641464
self.assertEqual(Outer.Inner.foo.value, 10)
14651465
self.assertEqual(
@@ -1525,7 +1525,7 @@ class Outer(Enum):
15251525
class Inner(Enum):
15261526
foo = 10
15271527
bar = 11
1528-
self.assertTrue(isinstance(Outer.Inner, Outer))
1528+
self.assertIsInstance(Outer.Inner, Outer)
15291529
self.assertEqual(Outer.a.value, 1)
15301530
self.assertEqual(Outer.Inner.value.foo.value, 10)
15311531
self.assertEqual(
@@ -2639,10 +2639,10 @@ def __new__(cls, value, period):
26392639
OneDay = day_1
26402640
OneWeek = week_1
26412641
OneMonth = month_1
2642-
self.assertFalse(hasattr(Period, '_ignore_'))
2643-
self.assertFalse(hasattr(Period, 'Period'))
2644-
self.assertFalse(hasattr(Period, 'i'))
2645-
self.assertTrue(isinstance(Period.day_1, timedelta))
2642+
self.assertNotHasAttr(Period, '_ignore_')
2643+
self.assertNotHasAttr(Period, 'Period')
2644+
self.assertNotHasAttr(Period, 'i')
2645+
self.assertIsInstance(Period.day_1, timedelta)
26462646
self.assertTrue(Period.month_1 is Period.day_30)
26472647
self.assertTrue(Period.week_4 is Period.day_28)
26482648

@@ -2719,13 +2719,13 @@ def _missing_(cls, item):
27192719
try:
27202720
Color('bad return')
27212721
except TypeError as exc:
2722-
self.assertTrue(isinstance(exc.__context__, ValueError))
2722+
self.assertIsInstance(exc.__context__, ValueError)
27232723
else:
27242724
raise Exception('Exception not raised.')
27252725
try:
27262726
Color('error out')
27272727
except ZeroDivisionError as exc:
2728-
self.assertTrue(isinstance(exc.__context__, ValueError))
2728+
self.assertIsInstance(exc.__context__, ValueError)
27292729
else:
27302730
raise Exception('Exception not raised.')
27312731

@@ -2864,7 +2864,7 @@ class ReformedColor(StrMixin, IntEnum, SomeEnum, AnotherEnum):
28642864
self.assertEqual(str(ReformedColor.BLUE), 'blue')
28652865
self.assertEqual(ReformedColor.RED.behavior(), 'booyah')
28662866
self.assertEqual(ConfusedColor.RED.social(), "what's up?")
2867-
self.assertTrue(issubclass(ReformedColor, int))
2867+
self.assertIsSubclass(ReformedColor, int)
28682868

28692869
def test_multiple_inherited_mixin(self):
28702870
@unique
@@ -2894,7 +2894,7 @@ class MyIntEnum(HexMixin, MyInt, enum.Enum):
28942894

28952895
class Foo(MyIntEnum):
28962896
TEST = 1
2897-
self.assertTrue(isinstance(Foo.TEST, MyInt))
2897+
self.assertIsInstance(Foo.TEST, MyInt)
28982898
self.assertEqual(Foo._member_type_, MyInt)
28992899
self.assertEqual(repr(Foo.TEST), "0x1")
29002900

@@ -2948,7 +2948,7 @@ class JobStatus(CaseInsensitiveStrEnum, LenientStrEnum):
29482948
self.assertEqual(JS.ACTIVE.value, 'active')
29492949
self.assertIs(JS('Active'), JS.ACTIVE)
29502950
self.assertTrue(JS.ACTIVE.valid)
2951-
self.assertTrue(isinstance(missing, JS))
2951+
self.assertIsInstance(missing, JS)
29522952
self.assertFalse(missing.valid)
29532953

29542954
def test_empty_globals(self):
@@ -3169,7 +3169,7 @@ class Entries(Foo, Enum):
31693169
ENTRY1 = 1
31703170
self.assertEqual(repr(Entries.ENTRY1), '<Entries.ENTRY1: ha hah!>')
31713171
self.assertTrue(Entries.ENTRY1.value == Foo(1), Entries.ENTRY1.value)
3172-
self.assertTrue(isinstance(Entries.ENTRY1, Foo))
3172+
self.assertIsInstance(Entries.ENTRY1, Foo)
31733173
self.assertTrue(Entries._member_type_ is Foo, Entries._member_type_)
31743174
#
31753175
# check auto-generated dataclass __repr__ is not used
@@ -3256,8 +3256,8 @@ class MyIntFlag(int, Flag):
32563256
ONE = 1
32573257
TWO = 2
32583258
FOUR = 4
3259-
self.assertTrue(isinstance(MyIntFlag.ONE | MyIntFlag.TWO, MyIntFlag), MyIntFlag.ONE | MyIntFlag.TWO)
3260-
self.assertTrue(isinstance(MyIntFlag.ONE | 2, MyIntFlag))
3259+
self.assertIsInstance(MyIntFlag.ONE | MyIntFlag.TWO, MyIntFlag, MyIntFlag.ONE | MyIntFlag.TWO)
3260+
self.assertIsInstance(MyIntFlag.ONE | 2, MyIntFlag)
32613261

32623262
def test_int_flags_copy(self):
32633263
class MyIntFlag(IntFlag):
@@ -4068,14 +4068,14 @@ def test_type(self):
40684068
self.assertTrue(Perm._member_type_ is int)
40694069
Open = self.Open
40704070
for f in Perm:
4071-
self.assertTrue(isinstance(f, Perm))
4071+
self.assertIsInstance(f, Perm)
40724072
self.assertEqual(f, f.value)
4073-
self.assertTrue(isinstance(Perm.W | Perm.X, Perm))
4073+
self.assertIsInstance(Perm.W | Perm.X, Perm)
40744074
self.assertEqual(Perm.W | Perm.X, 3)
40754075
for f in Open:
4076-
self.assertTrue(isinstance(f, Open))
4076+
self.assertIsInstance(f, Open)
40774077
self.assertEqual(f, f.value)
4078-
self.assertTrue(isinstance(Open.WO | Open.RW, Open))
4078+
self.assertIsInstance(Open.WO | Open.RW, Open)
40794079
self.assertEqual(Open.WO | Open.RW, 3)
40804080

40814081
@reraise_if_not_enum(HeadlightsK)

0 commit comments

Comments
 (0)