Remove dependency on unittest2 for the tests