Remove unittest2 dependency for tests