Migrate to unittest2