def test_input_files(patterns, recursive, size, sampling_rate, read_mode): Client.check_input(_input_files(patterns=patterns, recursive=recursive, size=size, sampling_rate=sampling_rate, read_mode=read_mode), data_type=DataInputType.CONTENT)
def test_mime_type(restful): f = Flow(restful=restful).add(uses='- !URI2Buffer {}') def validate_mime_type(req): for d in req.index.docs: assert d.mime_type == 'text/x-python' with f: f.index(_input_files('*.py'), validate_mime_type)
def index_generator(data_path, num_docs): for buffer in _input_files(data_path, True, num_docs, None, 'rb'): with Document() as doc: doc.buffer = buffer doc.mime_type = 'image/jpeg' yield doc
def test_input_files_with_invalid_read_mode(): with pytest.raises(BadClientInput): Client.check_input(_input_files(patterns='*.*', read_mode='invalid'))