Add assertions to compare images with reference ones