Add positional arguments to test.py