TDD with unittest