From ff09cd4b8f9a31b4224bd4b9d7be154323fb1d4d Mon Sep 17 00:00:00 2001
From: Rob Hallam <0504004h@student.gla.ac.uk>
Date: Thu, 22 Aug 2024 15:08:12 +0100
Subject: [PATCH] test: [mocks] add attrs to MockFeature +
 MockSource.duration()

---
 test/mocks.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

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