Test Driven Development - Python