adding the test python file