How can I write unit tests against code that uses matplotlib?

I’m working on a python (2.7) program that produce a lot of different matplotlib figure (the data are not random). I’m willing to implement some test (using unittest) to be sure that the generated figures are correct. For instance, I store the expected figure (data or image) in some place, I run my function and compare the result with the reference. Is there a way to do this ?