Unit Testing Example Python