updated py2 tests