|
|
@@ -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 |