Python Test Driven Development