test_matplotlib.py

import unittest
import os.path

import matplotlib.pyplot as plt
import numpy as np

class TestMatplotlib(unittest.TestCase):
    def test_plot(self):
        plt.plot(np.linspace(0,1,50), np.random.rand(50))
        plt.savefig("plot1.png")

        self.assertTrue(os.path.isfile("plot1.png"))