Unit testing with the unittest module