Update test_eshots.py