Explorar el Código

test: [mocks] add convenience classmethod and __eq__

Facilitates unit tests of TTA
main
Rob Hallam hace 1 mes
padre
commit
4e42137b33
Se han modificado 1 ficheros con 11 adiciones y 0 borrados
  1. +11
    -0
      test/mocks.py

+ 11
- 0
test/mocks.py Ver fichero

@@ -7,9 +7,16 @@ class MockInterval():
self.end = end
self.duration = end - start

@classmethod
def from_duration(cls, duration):
return cls(start=0, end=duration)

def to_json(self):
return {"start": self.start, "end": self.end}

def __eq__(self, other):
return self.start == other.start and self.end == other.end

class MockFeature():
"""Mock feature object for testing"""
def __init__(self, interval, source=None, feature_extractor="mock", score=0.0):
@@ -21,6 +28,10 @@ class MockFeature():
def to_json(self):
return {"interval": self.interval}

def __eq__(self, other):
return (self.interval == other.interval and self.source == other.source
and self.feature_extractor == other.feature_extractor)

class MockSource():
"""Mock Source object for testing Feature"""
def __init__(self, source=None, path=None):


Cargando…
Cancelar
Guardar