diff options
| author | Ethan Furman <ethan@stoneleaf.us> | 2020-09-19 18:12:57 (GMT) |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-09-19 18:12:57 (GMT) |
| commit | ae0d2a33ec05aece939a959d36fcf1df1e210a08 (patch) | |
| tree | b6a9ee1fe6ed468717a981a3ff33336ee921fa78 | |
| parent | 2b05361bf7cbbd76035206fd9befe87f37489f1e (diff) | |
| download | cpython-ae0d2a33ec05aece939a959d36fcf1df1e210a08.zip cpython-ae0d2a33ec05aece939a959d36fcf1df1e210a08.tar.gz cpython-ae0d2a33ec05aece939a959d36fcf1df1e210a08.tar.bz2 | |
bpo-41811: create SortKey members using first given value (GH-22316)
| -rw-r--r-- | Lib/pstats.py | 6 | ||||
| -rw-r--r-- | 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() |