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