In python3 use unittest instead of unittest2 which is used in python2