Added unittest.TestCase