Add a python test script