update tests to work with python 3