Update test_snake.py