Unit Testing Example Python Class