1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-27 01:48:17 +00:00
trezor-firmware/tests/test_trezor.utils.py
2017-12-16 02:11:26 +01:00

19 lines
527 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()