Adding python test.