diff options
| author | Raymond Hettinger <python@rcn.com> | 2010-12-03 02:09:34 +0000 |
|---|---|---|
| committer | Raymond Hettinger <python@rcn.com> | 2010-12-03 02:09:34 +0000 |
| commit | d8ff4658fb5884d9536732f7f34848aadd519e17 (patch) | |
| tree | 94c68d7d5582bbddcd56de6a74a742c5144f4bfd /Lib/test/test_itertools.py | |
| parent | a7a0e1a0f4248289e00284c115fff54b11f18a53 (diff) | |
| download | cpython-git-d8ff4658fb5884d9536732f7f34848aadd519e17.tar.gz | |
Simplify the signature for itertools.accumulate() to match numpy. Handle one item iterable the same way as min()/max().
Diffstat (limited to 'Lib/test/test_itertools.py')
| -rw-r--r-- | Lib/test/test_itertools.py | 16 |
1 files changed, 8 insertions, 8 deletions
diff --git a/Lib/test/test_itertools.py b/Lib/test/test_itertools.py index 8a67cff60c..b8f6eecbbe 100644 --- a/Lib/test/test_itertools.py +++ b/Lib/test/test_itertools.py @@ -59,18 +59,18 @@ class TestBasicOps(unittest.TestCase): def test_accumulate(self): self.assertEqual(list(accumulate(range(10))), # one positional arg - [0, 1, 3, 6, 10, 15, 21, 28, 36, 45]) - self.assertEqual(list(accumulate(range(10), 100)), # two positional args - [100, 101, 103, 106, 110, 115, 121, 128, 136, 145]) - self.assertEqual(list(accumulate(iterable=range(10), start=100)), # kw args - [100, 101, 103, 106, 110, 115, 121, 128, 136, 145]) + [0, 1, 3, 6, 10, 15, 21, 28, 36, 45]) + self.assertEqual(list(accumulate(iterable=range(10))), # kw arg + [0, 1, 3, 6, 10, 15, 21, 28, 36, 45]) for typ in int, complex, Decimal, Fraction: # multiple types - self.assertEqual(list(accumulate(range(10), typ(0))), + self.assertEqual( + list(accumulate(map(typ, range(10)))), list(map(typ, [0, 1, 3, 6, 10, 15, 21, 28, 36, 45]))) self.assertEqual(list(accumulate([])), []) # empty iterable - self.assertRaises(TypeError, accumulate, range(10), 0, 5) # too many args + self.assertEqual(list(accumulate([7])), [7]) # iterable of length one + self.assertRaises(TypeError, accumulate, range(10), 5) # too many args self.assertRaises(TypeError, accumulate) # too few args - self.assertRaises(TypeError, accumulate, range(10), x=7) # unexpected kwd args + self.assertRaises(TypeError, accumulate, x=range(10)) # unexpected kwd arg self.assertRaises(TypeError, list, accumulate([1, []])) # args that don't add def test_chain(self): |