Moved tests to python