diff options
Diffstat (limited to 'tests/__init__.py')
-rw-r--r-- | tests/__init__.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tests/__init__.py b/tests/__init__.py index 6acf0c99f..a32ca05d3 100644 --- a/tests/__init__.py +++ b/tests/__init__.py @@ -7,14 +7,15 @@ import unittest def main(): - tests = ["test_vercmp", "test_util"] + tests = ["test_atoms", "test_util"] suite = unittest.TestSuite() for mod in tests: try: - test_mod = __import__(mod) - suite.addTest(test_mod.suite()) + loadMod = __import__(mod) + tmpSuite = unittest.TestLoader().loadTestsFromModule(loadMod) + suite.addTest(tmpSuite) except ImportError: pass |