From b404a5050d533877eda66be51c63caef5cbcedff Mon Sep 17 00:00:00 2001
From: Andrew Chow <achow101-github@achow101.com>
Date: Mon, 14 Jan 2019 14:57:38 -0500
Subject: [PATCH] Include the device type in test case output

---
 test/test_device.py  | 6 ++++++
 test/test_keepkey.py | 6 ++++++
 test/test_trezor.py  | 6 ++++++
 3 files changed, 18 insertions(+)

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):