Adding python unit test