diff --git a/test/test_producers.py b/test/test_producers.py index 7d5f057..5f7d4ce 100644 --- a/test/test_producers.py +++ b/test/test_producers.py @@ -70,8 +70,16 @@ class TestVisualisationProducer(unittest.TestCase): def test_produce(self): """Test we can call produce without error""" - # TODO - pass + # to do this, we need to mock out plt.savefig(path) + feature = mocks.MockFeature(interval=mocks.MockInterval(start=0, end=5), + source=mocks.MockSource(path="test.mp4")) + features = [feature] + visprod = producers.VisualisationProducer(features=features) + with unittest.mock.patch("matplotlib.pyplot.savefig") as mock_savefig: + visprod.produce() + mock_savefig.assert_called() + + # TODO: more more granular test coverage of produce behaviour class TestJSONProducer(unittest.TestCase): def test_init_empty(self):