1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-10-11 02:19:21 +00:00
trezor-firmware/tests/test_trezor.utils.py
2016-11-06 17:04:25 +01:00

17 lines
525 B
Python

from common import *
from trezor import utils
class TestUtils(unittest.TestCase):
def test_chunks(self):
c = list(utils.chunks(range(100), 7))
for i in range(15):
# need to check start, stop, step attrs until https://github.com/micropython/micropython/issues/2600 is resolved
self.assertEqual(c[i].start, i * 7)
self.assertEqual(c[i].stop, 100 if (i == 14) else (i + 1) * 7)
self.assertEqual(c[i].step, 1)
if __name__ == '__main__':
unittest.main()