diff --git a/tests/test_trezor.msg.py b/tests/test_trezor.msg.py new file mode 100644 index 0000000000..dd45145baf --- /dev/null +++ b/tests/test_trezor.msg.py @@ -0,0 +1,19 @@ +from common import * + +from trezor.crypto import random + +from trezor import msg + +class TestMsg(unittest.TestCase): + + def test_set_get_interfaces(self): + ifaces = msg.get_interfaces() + self.assertEqual(ifaces, ()) + for n in range(1, 9): + ifaces = tuple((random.uniform(0x10000) for _ in range(n))) + msg.set_interfaces(ifaces) + ifaces2 = msg.get_interfaces() + self.assertEqual(ifaces, ifaces2) + +if __name__ == '__main__': + unittest.main() diff --git a/tests/test_trezor.ui.display.py b/tests/test_trezor.ui.display.py new file mode 100644 index 0000000000..a8f6a51111 --- /dev/null +++ b/tests/test_trezor.ui.display.py @@ -0,0 +1,72 @@ +from common import * + +from trezor.crypto import random + +from trezor.ui import display + +class TestDisplay(unittest.TestCase): + + def test_clear(self): + display.clear() + + def test_refresh(self): + display.refresh() + + def test_bar(self): + display.bar(0, 0, 10, 10, 0xFFFF) + + def test_bar_radius(self): + display.bar_radius(0, 0, 10, 10, 0xFFFF, 0x0000, 16) + + def test_image(self): + pass + + def test_icon(self): + pass + + def test_text(self): + display.text(120, 120, 'Test', 0, 0xFFFF, 0x0000) + + def test_text_center(self): + display.text_center(120, 120, 'Test', 0, 0xFFFF, 0x0000) + + def test_text_right(self): + display.text_right(120, 120, 'Test', 0, 0xFFFF, 0x0000) + + def test_text_width(self): + display.text_width('Test', 0) + + def test_qrcode(self): + display.qrcode(0, 0, 'Test', 4) + + def test_loader(self): + display.loader(333, 0, 0xFFFF, 0x0000) + + def test_orientation(self): + for o in [0, 90, 180, 270]: + display.orientation(o) + o2 = display.orientation() + self.assertEqual(o, o2) + + def test_backlight(self): + for b in range(256): + display.backlight(b) + b2 = display.backlight() + self.assertEqual(b, b2) + + def test_offset(self): + for x in range(-4, 5): + for y in range(-4, 5): + o = (x * 57, y * 57) + display.offset(o) + o2 = display.offset() + self.assertEqual(o, o2) + + def test_raw(self): + pass + + def test_save(self): + pass + +if __name__ == '__main__': + unittest.main()