mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-26 09:28:13 +00:00
29 lines
834 B
Python
29 lines
834 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)
|
|
|
|
def test_format_amount(self):
|
|
VECTORS = [
|
|
(123456, 3, "123.456"),
|
|
(4242, 7, "0.0004242"),
|
|
(-123456, 3, "-123.456"),
|
|
(-4242, 7, "-0.0004242"),
|
|
]
|
|
for v in VECTORS:
|
|
self.assertEqual(utils.format_amount(v[0], v[1]), v[2])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|