Unit Testing Example Python Loop