def test_mime_type():
f = Flow().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 test_input_files(patterns, recursive, size, sampling_rate, read_mode):
PyClient.check_input(
input_files(
patterns=patterns,
recursive=recursive,
size=size,
sampling_rate=sampling_rate,
read_mode=read_mode
)
)
def test_io_files(self):
PyClient.check_input(input_files('*.*'))
PyClient.check_input(input_files('*.*', recursive=True))
PyClient.check_input(input_files('*.*', size=2))
PyClient.check_input(input_files('*.*', size=2, read_mode='rb'))
PyClient.check_input(input_files('*.*', sampling_rate=.5))
f = Flow().add(uses='- !URI2Buffer {}')
def validate_mime_type(req):
for d in req.index.docs:
self.assertEqual(d.mime_type, 'text/x-python')
with f:
f.index(input_files('*.py'), validate_mime_type)
def test_input_files_with_invalid_read_mode():
with pytest.raises(RuntimeError):
PyClient.check_input(input_files(patterns='*.*', read_mode='invalid'))
def test_io_files():
PyClient.check_input(input_files('*.*'))
PyClient.check_input(input_files('*.*', recursive=True))
PyClient.check_input(input_files('*.*', size=2))
PyClient.check_input(input_files('*.*', size=2, read_mode='rb'))
PyClient.check_input(input_files('*.*', sampling_rate=0.5))
