Plot testing

Finally a way to test plotting functions, using matplotlib's testing function `compare_images`.