Update unit tests to verify matplotlib output formats