add tests to python