From 3e9a071926bb5a15eec0715a350ffd6d9246e644 Mon Sep 17 00:00:00 2001
From: Shlomi Fish <shlomif@shlomifish.org>
Date: Mon, 19 Mar 2018 15:46:26 +0200
Subject: [PATCH] convert to unittest

---
 tests/unit/hint.py | 37 +++++++++++++++++--------------------
 1 file changed, 17 insertions(+), 20 deletions(-)

diff --git a/tests/unit/hint.py b/tests/unit/hint.py
index 258f13ef..bb9e3440 100644
--- a/tests/unit/hint.py
+++ b/tests/unit/hint.py
@@ -1,31 +1,28 @@
 #!/usr/bin/env python3
 # Written by Shlomi Fish, under the MIT Expat License.
 
-# imports
-import sys
-from TAP.Simple import diag, plan, ok
-
+import unittest
 from pysollib.acard import AbstractCard
 from pysollib.hint import Base_Solver_Hint
 
 
-def shlomif_main(args):
+class MyTests(unittest.TestCase):
+    def test_output(self):
+        card = AbstractCard(1001, 0, 3, 7, 3001)
+        h = Base_Solver_Hint(None, None, base_rank=0)
 
-    plan(2)
+        got = h.card2str2(card)
+        # TEST
+        self.assertEqual(got, 'D-8', 'card2str2 works')
+        # diag('got == ' + got)
 
-    card = AbstractCard(1001, 0, 3, 7, 3001)
-    h = Base_Solver_Hint(None, None, base_rank=0)
-
-    got = h.card2str2(card)
-    # TEST
-    if not ok(got == 'D-8', 'card2str2 works'):
-        diag('got == ' + got)
-
-    got = h.card2str1(card)
-    # TEST
-    if not ok(got == '8D', 'card2str1 works'):
-        diag('got == ' + got)
+        got = h.card2str1(card)
+        # TEST
+        self.assertEqual(got, '8D', 'card2str2 works')
+        # diag('got == ' + got)
 
 
-if __name__ == "__main__":
-    sys.exit(shlomif_main(sys.argv))
+if __name__ == '__main__':
+    from pycotap import TAPTestRunner
+    suite = unittest.TestLoader().loadTestsFromTestCase(MyTests)
+    TAPTestRunner().run(suite)