def __init__(self, args):
self.args = args
self.video_src = VideoSource(str(args.src.absolute()),
backend=args.opencv_videoio_backend)
self.video_sink = get_flow_writer(args)
self.flow_algorithm = self.get_flow_algorithm(args)
def test_bound_set_in_uv_flow_writer(self):
bound = 25
writer = get_flow_writer(self.create_args('jpg', bound=bound))
assert isinstance(writer, FlowUVImageWriter)
assert writer.bound == bound
def test_raises_error_on_unknown_extension(self):
with pytest.raises(ValueError):
get_flow_writer(self.create_args("asdf"))
def test_flo_returns_middlebury_flow_writer(self):
writer = get_flow_writer(self.create_args("flo"))
assert isinstance(writer, MiddleburyFlowWriter)
def test_npy_returns_flow_numpy_writer(self, extension):
writer = get_flow_writer(self.create_args(extension))
assert isinstance(writer, FlowNumpyWriter)
def test_image_extensions_returns_flow_uv_writer(self, extension):
writer = get_flow_writer(self.create_args(extension, bound=25))
assert isinstance(writer, FlowUVImageWriter)