Importing only class TestCase instead of the complete unittest module