Update image_test.py