Use unittest2 instead of unittest