Using the unittest module