1 files changed, 3 insertions, 4 deletions

diff --git a/test.py b/test.py
index a87b2e0..c3fad8e 100755
--- a/test.py
+++ b/test.py
@@ -1,7 +1,6 @@
 #!/usr/bin/env python3
-from doctest import testmod
-from pyinflux import client, parser
+import unittest
+from pyinflux import client, parser, test
 
 if __name__ == '__main__':
-    testmod(m=client)
-    testmod(m=parser)
+    unittest.TextTestRunner().run(unittest.findTestCases(test))
\ No newline at end of file