diff --git a/test/test_producers.py b/test/test_producers.py index 0680b59..c95e706 100644 --- a/test/test_producers.py +++ b/test/test_producers.py @@ -10,6 +10,34 @@ class MockFeature(): def to_json(self): return {"interval": self.interval} +class TestFfmpegVideoProducer(unittest.TestCase): + """Test FfmpegVideoProducer (produces videos using ffmpeg)""" + + def test_init_empty(self): + # test with no features -- should raise ValueError + with self.assertRaises(ValueError): + producers.FfmpegVideoProducer(features=None) + + def test_init(self): + # test with features + feature = MockFeature(interval="") + features = [feature] + vidprod = producers.FfmpegVideoProducer(features=features) + self.assertEqual(vidprod.features, features) + + def test_ffmpeg_feature_to_clip(self): + # TODO + pass + + def test_ffmpeg_concat_clips(self): + # TODO + pass + + def test_produce(self): + """Test we can call produce without error""" + # TODO + pass + class TestVisualisationProducer(unittest.TestCase): """Test VisualisationProducer (produces matplotlib visualisations)"""