From ae0d2a33ec05aece939a959d36fcf1df1e210a08 Mon Sep 17 00:00:00 2001 From: Ethan Furman Date: Sat, 19 Sep 2020 11:12:57 -0700 Subject: bpo-41811: create SortKey members using first given value (GH-22316) --- Lib/pstats.py | 6 +++--- Lib/test/test_pstats.py | 4 ++++ 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/Lib/pstats.py b/Lib/pstats.py index e781b91..0f93ae0 100644 --- a/Lib/pstats.py +++ b/Lib/pstats.py @@ -45,9 +45,9 @@ class SortKey(str, Enum): TIME = 'time', 'tottime' def __new__(cls, *values): - obj = str.__new__(cls) - - obj._value_ = values[0] + value = values[0] + obj = str.__new__(cls, value) + obj._value_ = value for other_value in values[1:]: cls._value2member_map_[other_value] = obj obj._all_values = values diff --git a/Lib/test/test_pstats.py b/Lib/test/test_pstats.py index 10559de..4f78b99 100644 --- a/Lib/test/test_pstats.py +++ b/Lib/test/test_pstats.py @@ -95,5 +95,9 @@ class StatsTestCase(unittest.TestCase): self.assertIn('pass2', funcs_called) self.assertIn('pass3', funcs_called) + def test_SortKey_enum(self): + self.assertEqual(SortKey.FILENAME, 'filename') + self.assertNotEqual(SortKey.FILENAME, SortKey.CALLS) + if __name__ == "__main__": unittest.main() -- cgit v0.12