add tests file
add proper output checker for python 2 tests