40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
import unittest
|
|
|
|
from accumulate import accumulate
|
|
|
|
|
|
class AccumulateTest(unittest.TestCase):
|
|
def test_empty_sequence(self):
|
|
self.assertEqual(accumulate([], lambda x: x / 2), [])
|
|
|
|
def test_pow(self):
|
|
self.assertEqual(
|
|
accumulate([1, 2, 3, 4, 5], lambda x: x * x), [1, 4, 9, 16, 25])
|
|
|
|
def test_divmod(self):
|
|
self.assertEqual(
|
|
accumulate([10, 17, 23], lambda x: divmod(x, 7)),
|
|
[(1, 3), (2, 3), (3, 2)])
|
|
|
|
def test_composition(self):
|
|
inp = [10, 17, 23]
|
|
self.assertEqual(
|
|
accumulate(
|
|
accumulate(inp, lambda x: divmod(x, 7)),
|
|
lambda x: 7 * x[0] + x[1]), inp)
|
|
|
|
def test_capitalize(self):
|
|
self.assertEqual(
|
|
accumulate(['hello', 'world'], str.upper), ['HELLO', 'WORLD'])
|
|
|
|
def test_recursive(self):
|
|
inp = ['a', 'b', 'c']
|
|
out = [['a1', 'a2', 'a3'], ['b1', 'b2', 'b3'], ['c1', 'c2', 'c3']]
|
|
self.assertEqual(
|
|
accumulate(
|
|
inp, lambda x: accumulate(list('123'), lambda y: x + y)), out)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|