tests: support python 3 in tests