Update test_es.py