unit tests working in both python2.7 and python3