Replace unittest with unittest2