def test_processor_properties():
vision = VisionSubclass("Test")
processorA = ProcessorA(vision)
processorB = ProcessorB(processorA)
with processorB as s:
assert(s.autoexposure is None)
assert(s.autofocus is None)
assert(s.autowhitebalance is None)
assert(s.autogain is None)
assert(s.exposure is None)
assert(s.focus is None)
assert(s.whitebalance is None)
s.autoexposure = 1
s.autofocus = 2
s.autowhitebalance = 3
s.autogain = 4
s.exposure = 5
s.focus = 6
s.whitebalance = 7
assert(vision.autoexposure == 1)
assert(vision.autofocus == 2)
assert(vision.autowhitebalance == 3)
assert(vision.autogain == 4)
assert(vision.exposure == 5)
assert(vision.focus == 6)
assert(vision.whitebalance == 7)
def test_capture_disabled():
vision = VisionSubclass(0)
with ProcessorA(vision, enabled=False) as processor:
img = processor.capture()
assert(isinstance(img, Frame))
assert(img.images[0].source is vision)
assert(img.images[0].image == "an image")
def test_capture():
vision = VisionSubclass(0)
with ProcessorA(vision) as processor:
img = processor.capture()
assert(isinstance(img, Frame))
assert(img.images[0].source is processor)
assert(img.images[0].image == "AN IMAGE")
def test_capture_stacked_incorrect():
vision = VisionSubclass("Test")
processorA = ProcessorA(vision)
processorB = ProcessorB(processorA)
assert(processorB.name == "ProcessorB <- ProcessorA <- Test")
with raises(AssertionError):
processorB.capture()
def test_capture_mask_processor_override():
vision = VisionSubclass(0, num_images=2, processor_mask="10")
with ProcessorA(vision, processor_mask="01") as processor:
img = processor.capture()
assert(isinstance(img, Frame))
assert(img.images[0].source is vision)
assert(img.images[0].image == "an image")
assert(img.images[1].source is processor)
assert(img.images[1].image == "AN IMAGE1")
def test_method_resolution():
vision = VisionSubclass("Test")
processorA = ProcessorA(vision)
processorB = ProcessorB(processorA)
assert(processorB.name == "ProcessorB <- ProcessorA <- Test")
assert(not vision.camera_called)
assert(processorB.camera_())
assert(processorB._camera_called)
assert(vision._camera_called)
def test_capture_stacked():
vision = VisionSubclass("Test")
processorA = ProcessorA(vision)
processorB = ProcessorB(processorA)
assert(processorB.name == "ProcessorB <- ProcessorA <- Test")
with processorB as processor:
img = processor.capture()
assert(isinstance(img, Frame))
assert(img.images[0].source is processorB)
assert(img.images[0].image == "An Image")
assert(processorB.get_source('VisionSubclass') is vision)
assert(processorB.get_source('ProcessorA') is processorA)
assert(processorB.get_source('ProcessorB') is processorB)
assert(processorB.get_source('Test no') is None)
def test_implementation():
source = VisionSubclass()
pr = ProcessorA(source)
assert(pr.source is source)
def test_capture_incorrect():
vision = VisionSubclass(0)
processor = ProcessorA(vision)
with raises(AssertionError):
processor.capture()