1
0

util: add function "chunks"

taken from ElectrumX
67111a3c4c/electrumx/lib/util.py (L149)
This commit is contained in:
SomberNight
2019-05-26 04:10:32 +02:00
parent 1ebfcc0f36
commit 30ffb3d4dc
2 changed files with 15 additions and 1 deletions

View File

@@ -1,7 +1,7 @@
from decimal import Decimal
from electrum.util import (format_satoshis, format_fee_satoshis, parse_URI,
is_hash256_str)
is_hash256_str, chunks)
from . import SequentialTestCase
@@ -104,3 +104,9 @@ class TestUtil(SequentialTestCase):
self.assertFalse(is_hash256_str('qweqwe'))
self.assertFalse(is_hash256_str(None))
self.assertFalse(is_hash256_str(7))
def test_chunks(self):
self.assertEqual([[1, 2], [3, 4], [5]],
list(chunks([1, 2, 3, 4, 5], 2)))
with self.assertRaises(ValueError):
list(chunks([1, 2, 3], 0))