Learning TDD in python