diff options
Diffstat (limited to '')
-rw-r--r-- | test/unit.py | 8 |
1 files 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) |