Writing tests for your module using Python unit tests