From 4961e6814709bb5eab9b82a887900722249146df Mon Sep 17 00:00:00 2001
From: Stefan Behnel <stefan_ml@behnel.de>
Date: Mon, 14 Mar 2022 12:55:55 +0100
Subject: [PATCH] Avoid deprecated test loader function.

Closes https://github.com/scoder/fastrlock/issues/16
---
 fastrlock/tests/__init__.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/fastrlock/tests/__init__.py b/fastrlock/tests/__init__.py
index 68aebb7..22f5c9e 100644
--- 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