Software Test Driven Python Code