Updates to image testing framework

Implement changes made by @sargas in WCSAxes (since we use the same image testing framework here)