Fix Python/test.sh script to work with new style test