Unit testing your code with the unittest module