Update test_mars.py