Look at the visualization code and figure out how to test it. Check which functions can be tested by regular unit testing and where we need more sophisticated techniques like image diff.