Sfoglia il codice sorgente

test: add MockSource & MockInterval to test Feature

main
Rob Hallam 2 mesi fa
parent
commit
0a08af4972
1 ha cambiato i file con 22 aggiunte e 0 eliminazioni
  1. +22
    -0
      test/test_utils.py

+ 22
- 0
test/test_utils.py Vedi 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


Caricamento…
Annulla
Salva