Unit Testing Python Example