Tests: class prefix made optional.

2018-11-15 21:26:15 +03:00
parent eff760bd2b
commit 41d3d63758
1 changed files with 6 additions and 2 deletions
--- a/test/unit.py
+++ b/test/unit.py
@@ -31,10 +31,14 @@ class TestUnit(unittest.TestCase):

            TestUnit._set_args(args)

-    @staticmethod
-    def main():
+    @classmethod
+    def main(cls):
        args, rest = TestUnit._parse_args()

+        for i, arg in enumerate(rest):
+            if arg[:5] == 'test_':
+                rest[i] = cls.__name__ + '.' + arg
+
        sys.argv = sys.argv[:1] + rest

        TestUnit._set_args(args)