Add command line args to python test script.