Added some more tests using python instead of ruby