Add some image comparison tests

Created by: Zaharid

I tested this:

https://matplotlib.org/devel/testing.html#writing-an-image-comparison-test

and basically it worked out of the box. Maybe would be good to have it at least for some plots.