refactored tests and suite

3 files changed, 17 insertions, 2 deletions

diff --git a/dbutils_test.py b/dbutils_test.py
index 233e536..3289264 100644
--- a/dbutils_test.py
+++ b/dbutils_test.py
@@ -9,4 +9,5 @@ class DBUtilsTest(unittest.TestCase):
         self.assertTrue(len(tables) >= 1)
 
 if __name__ == '__main__':
-    unittest.main()
+    suite = unittest.TestLoader().loadTestsFromTestCase(DBUtilsTest)
+    unittest.TextTestRunner(verbosity=2).run(suite)
diff --git a/runtests.py b/runtests.py
new file mode 100644
index 0000000..c5b22e1
--- /dev/null
+++ b/runtests.py
@@ -0,0 +1,13 @@
+import unittest
+
+from dbutils_test import DBUtilsTest
+from sensors_test import SensorsTest
+
+def createSuite():
+    suite = unittest.TestSuite()
+    suite.addTests(unittest.TestLoader().loadTestsFromTestCase(DBUtilsTest))
+    suite.addTests(unittest.TestLoader().loadTestsFromTestCase(SensorsTest))
+    return suite
+
+suite = createSuite()
+unittest.TextTestRunner(verbosity=2).run(suite)
diff --git a/sensors_test.py b/sensors_test.py
index cc30b44..f82e757 100644
--- a/sensors_test.py
+++ b/sensors_test.py
@@ -9,4 +9,5 @@ class SensorsTest(unittest.TestCase):
         self.assertTrue(len(sen) >= 1)
 
 if __name__ == '__main__':
-    unittest.main()
+    suite = unittest.TestLoader().loadTestsFromTestCase(SensorsTest)
+    unittest.TextTestRunner(verbosity=2).run(suite)