Python testing goat code