Extend unittest.TestCase