Update tests for python 3