Extending the unittest would be a good idea.