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'))
