test: [VizProd] simple colour test

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