Adding python unit tests