Adding image unit tests