From 41d3d63758fc3846d5a09afd3b33aac19231942a Mon Sep 17 00:00:00 2001 From: Andrey Zelenkov Date: Thu, 15 Nov 2018 21:26:15 +0300 Subject: [PATCH] Tests: class prefix made optional. --- test/unit.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/test/unit.py b/test/unit.py index 0415d83a..c3efef26 100644 --- 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)