Update test_getentity.py