Преглед на файлове

test: add MockSource & MockInterval to test Feature

main
Rob Hallam преди 3 месеца
родител
ревизия
0a08af4972
променени са 1 файла, в които са добавени 22 реда и са изтрити 0 реда
  1. +22
    -0
      test/test_utils.py

+ 22
- 0
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


Зареждане…
Отказ
Запис