now working with python tests (some memory issues though)