Unit Testing Example Python Program