Move tests over from ruby to python