Test driven development in python