diff options
| author | Serhiy Storchaka <storchaka@gmail.com> | 2019-09-10 16:31:01 +0300 |
|---|---|---|
| committer | Dino Viehland <dinoviehland@gmail.com> | 2019-09-10 14:31:01 +0100 |
| commit | 88bdb9280b251d753f1b1c8d9183de0fff003622 (patch) | |
| tree | c8a96f484735cc1757ceda1aabfb14df86ea3fcf /Lib/test/test_builtin.py | |
| parent | c1d8c1cb8e90a54a3daaa7fcdb8d6ca7f08d6a73 (diff) | |
| download | cpython-88bdb9280b251d753f1b1c8d9183de0fff003622.tar.gz cpython-88bdb9280b251d753f1b1c8d9183de0fff003622.zip | |
bpo-36781: Optimize sum() for bools. (#13074)
* Optimize sum() for bools.
* Fix sum([], False).
* Add a NEWS entry.
Diffstat (limited to 'Lib/test/test_builtin.py')
| -rw-r--r-- | Lib/test/test_builtin.py | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/Lib/test/test_builtin.py b/Lib/test/test_builtin.py index db2f6cd49b2..7b3c054d5eb 100644 --- a/Lib/test/test_builtin.py +++ b/Lib/test/test_builtin.py @@ -1373,6 +1373,24 @@ class BuiltinTest(unittest.TestCase): self.assertEqual(sum(range(10), 1000), 1045) self.assertEqual(sum(range(10), start=1000), 1045) + self.assertEqual(sum(range(10), 2**31-5), 2**31+40) + self.assertEqual(sum(range(10), 2**63-5), 2**63+40) + + self.assertEqual(sum(i % 2 != 0 for i in range(10)), 5) + self.assertEqual(sum((i % 2 != 0 for i in range(10)), 2**31-3), + 2**31+2) + self.assertEqual(sum((i % 2 != 0 for i in range(10)), 2**63-3), + 2**63+2) + self.assertIs(sum([], False), False) + + self.assertEqual(sum(i / 2 for i in range(10)), 22.5) + self.assertEqual(sum((i / 2 for i in range(10)), 1000), 1022.5) + self.assertEqual(sum((i / 2 for i in range(10)), 1000.25), 1022.75) + self.assertEqual(sum([0.5, 1]), 1.5) + self.assertEqual(sum([1, 0.5]), 1.5) + self.assertEqual(repr(sum([-0.0])), '0.0') + self.assertEqual(repr(sum([-0.0], -0.0)), '-0.0') + self.assertEqual(repr(sum([], -0.0)), '-0.0') self.assertRaises(TypeError, sum) self.assertRaises(TypeError, sum, 42) @@ -1384,6 +1402,9 @@ class BuiltinTest(unittest.TestCase): self.assertRaises(TypeError, sum, [[1], [2], [3]]) self.assertRaises(TypeError, sum, [{2:3}]) self.assertRaises(TypeError, sum, [{2:3}]*2, {2:3}) + self.assertRaises(TypeError, sum, [], '') + self.assertRaises(TypeError, sum, [], b'') + self.assertRaises(TypeError, sum, [], bytearray()) class BadSeq: def __getitem__(self, index): |