Simple TDD practise in Python