Avoid deprecated test loader function.

Closes #16
scoder · Mar 14, 2022 · 4961e68 · 4961e68
1 parent dc77e21
commit 4961e68
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fastrlock/tests/__init__.py b/fastrlock/tests/__init__.py
@@ -8,9 +8,9 @@ def suite():
     tests = []
     for filename in os.listdir(test_dir):
         if filename.endswith('.py') and not filename.startswith('_'):
-            tests.append(unittest.findTestCases('tests.'+ filename[:-3]))
+            tests.append(unittest.defaultTestLoader.loadTestsFromName(__name__ + '.' + filename[:-3]))
 
-    suite = unittest.TestSuite(tests)
+    suite = unittest.TestSuite(tests) if len(tests) != 1 else tests[0]
     suite.addTest(doctest.DocFileSuite('../../README.rst'))
     return suite