Add python script to run tests