Use unittest2.TestCase