Browse Source

test: add MockSource & MockInterval to test Feature

main
Rob Hallam 3 months ago
parent
commit
0a08af4972
1 changed files with 22 additions and 0 deletions
  1. +22
    -0
      test/test_utils.py

+ 22
- 0
test/test_utils.py View File

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


Loading…
Cancel
Save