diff --git a/test/test_utils.py b/test/test_utils.py index 4876366..460ad3c 100644 --- a/test/test_utils.py +++ b/test/test_utils.py @@ -3,6 +3,28 @@ import unittest import pipeline.utils as utils +class MockSource(): + """Mock Source object for testing Feature""" + def __init__(self, source): + self.source = source + + def __str__(self): + return self.source + + def to_json(self): + return {"source": self.source} + + def __eq__(self, other): + return self.source == other.source + +class MockInterval(): + """Mock Interval object for testing Feature""" + def __init__(self, start, end): + self.start = start + self.end = end + + def to_json(self): + return {"start": self.start, "end": self.end} class TestSource(unittest.TestCase): """Source is a container for source, path, and provider of a media file