diff --git a/test/test_producers.py b/test/test_producers.py index d0af2cf..a9c4ead 100644 --- a/test/test_producers.py +++ b/test/test_producers.py @@ -81,6 +81,21 @@ class TestVisualisationProducer(unittest.TestCase): # TODO: more more granular test coverage of produce behaviour + def test_fe_colour(self): + """test that colour picker returns the right colours# + + laughter=red, loudness=blue, video activity=green, words=yellow + """ + # this is a pretty poor test really + expected = {"laughter": "red", "loudness": "blue", "videoactivity": "green", "words": "purple"} + feature = mocks.MockFeature(interval="") + features = [feature] + visprod = producers.VisualisationProducer(features=features) + + for given, expected in expected.items(): + feature.feature_extractor = given + self.assertEqual(visprod._fe_colour(feature), expected) + class TestJSONProducer(unittest.TestCase): def test_init_empty(self): # test with no features -- should raise ValueError