Update test_get_image.py

Refactored tests.