Support py2 and py3 tests