Python 3 TDD example