diff --git a/test/mocks.py b/test/mocks.py
index a9153e9..4cb016c 100644
--- a/test/mocks.py
+++ b/test/mocks.py
@@ -12,9 +12,11 @@ class MockInterval():
 
 class MockFeature():
     """Mock feature object for testing"""
-    def __init__(self, interval, source=None):
+    def __init__(self, interval, source=None, feature_extractor="mock", score=0.0):
         self.interval = interval
         self.source = source
+        self.feature_extractor = feature_extractor
+        self.score = score
 
     def to_json(self):
         return {"interval": self.interval}
@@ -33,3 +35,7 @@ class MockSource():
 
     def __eq__(self, other):
         return self.source == other.source
+
+    def duration(self, duration=30.0):
+        """Mock a float duration"""
+        return duration