diff --git a/test/test_device.py b/test/test_device.py index cd28990..462a606 100644 --- a/test/test_device.py +++ b/test/test_device.py @@ -75,6 +75,12 @@ class DeviceTestCase(unittest.TestCase): suite.addTest(testclass(rpc, rpc_userpass, type, path, fingerprint, master_xpub, password, emulator, name)) return suite + def __str__(self): + return '{}: {}'.format(self.type, super().__str__()) + + def __repr__(self): + return '{}: {}'.format(self.type, super().__repr__()) + class TestDeviceConnect(DeviceTestCase): def setUp(self): self.emulator.start() diff --git a/test/test_keepkey.py b/test/test_keepkey.py index 53da65a..748257c 100755 --- a/test/test_keepkey.py +++ b/test/test_keepkey.py @@ -73,6 +73,12 @@ class KeepkeyTestCase(unittest.TestCase): suite.addTest(testclass(emulator, name)) return suite + def __str__(self): + return 'keepkey: {}'.format(super().__str__()) + + def __repr__(self): + return 'keepkey: {}'.format(super().__repr__()) + # Keepkey specific getxpub test because this requires device specific thing to set xprvs class TestKeepkeyGetxpub(KeepkeyTestCase): def setUp(self): diff --git a/test/test_trezor.py b/test/test_trezor.py index 06d51f4..dd7be0c 100755 --- a/test/test_trezor.py +++ b/test/test_trezor.py @@ -70,6 +70,12 @@ class TrezorTestCase(unittest.TestCase): suite.addTest(testclass(emulator, name)) return suite + def __str__(self): + return 'trezor: {}'.format(super().__str__()) + + def __repr__(self): + return 'trezor: {}'.format(super().__repr__()) + # Trezor specific getxpub test because this requires device specific thing to set xprvs class TestTrezorGetxpub(TrezorTestCase): def setUp(self):