mediawiki_dump_tools/Mediawiki-Utilities/mw/util/iteration/tests/test_aggregate.py

   1 from nose.tools import eq_
   2 from ..aggregate import aggregate
   3
   4
   5 def test_group():
   6     l = [0, 1, 2, 3, 4, 5, 10, 11, 12, 13, 14]
   7     expected = [[0, 1, 2, 3, 4, 5], [10, 11, 12, 13, 14]]
   8
   9     result = []
  10     for identifier, group in aggregate(l, lambda item: int(item / 10)):
  11         result.append(list(group))
  12
  13     eq_(result, expected)