Python program to run tests