|
@@ -10,6 +10,34 @@ class MockFeature(): |
|
|
def to_json(self): |
|
|
def to_json(self): |
|
|
return {"interval": self.interval} |
|
|
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): |
|
|
class TestVisualisationProducer(unittest.TestCase): |
|
|
"""Test VisualisationProducer (produces matplotlib visualisations)""" |
|
|
"""Test VisualisationProducer (produces matplotlib visualisations)""" |
|
|
|
|
|
|
|
|