def shim():
model = PythonStaticMethodModel.from_function_pointer(run)
model.set_name('test')
model.set_title('Test')
model.set_inputs('bool', 'Whether to run successfully')
model.set_outputs('bool', 'Should always be True')
return PythonStaticMethodServable(**model.to_dict())
def test_multiargs():
# Make the maximum function
model = PythonStaticMethodModel.from_function_pointer(max) \
.set_name('test').set_title('test')
# Describe the inputs
model.set_inputs('tuple', 'Two numbers',
element_types=[
compose_argument_block('float', 'A number'),
compose_argument_block('float', 'A second number')
])
model.set_outputs('float', 'Maximum of the two numbers')
model.set_unpack_inputs(True)
# Make sure the shim works
servable = PythonStaticMethodServable(**model.to_dict())
assert servable.run((1, 2))[0] == 2
def test_static():
# Make an example static method
model = PythonStaticMethodModel.create_model('numpy', 'max', autobatch=False,
function_kwargs={'axis': 0})
model.set_title('Example function')
model.set_name('function')
model.set_inputs('ndarray', 'Matrix', shape=[None, None])
model.set_outputs('ndarray', 'Max of a certain axis', shape=[None])
# Make the servable
servable = PythonStaticMethodServable(**model.to_dict())
# Test it
out, _ = servable.run([[1, 2], [3, 4]])
assert np.isclose([3, 4], out).all()
# Test giving it parameters
assert np.isclose([2, 4], servable.run([[1, 2], [3, 4]], parameters=dict(axis=1))[0]).all()
# Test the autobatch
model.servable.methods['run'].method_details['autobatch'] = True
servable = PythonStaticMethodServable(**model.to_dict())
out, _ = servable.run([[1, 2], [3, 4]])
assert np.isclose([2, 4], out).all()
def test_single_file_input(tmpdir):
# Make the metadata model
model = PythonStaticMethodModel.from_function_pointer(os.path.isfile).set_name('test')
model.set_title('test')
model.set_inputs('file', 'A file')
model.set_outputs('boolean', 'Whether it exists')
# Make the servable
servable = PythonStaticMethodServable(**model.to_dict())
# Run on local file
assert servable.run({'url': __file__})[0]
if system() != 'Windows':
assert servable.run({'url': 'file:///' + __file__})[0]
# Run on remote file
assert servable.run({
'url': 'https://www.google.com/images/branding/googlelogo/1x/googlelogo_color_272x92dp.png'}
)[0]
def test_single_file_list_input():
# Make the metadata model
model = PythonStaticMethodModel.from_function_pointer(os.path.isfile, autobatch=True)
model.set_name('test')
model.set_title('test')
model.set_inputs('list', 'List of files', item_type='file')
model.set_outputs('list', 'Whether each file exists', item_type='boolean')
# Make the servable
servable = PythonStaticMethodServable(**model.to_dict())
# Run on local file
assert servable.run([{'url': __file__}])[0]
if system() != 'Windows':
assert servable.run([{'url': 'file:///' + __file__}])[0] # Fail on Windows
# Run on remote file
assert servable.run([{
'url': 'https://www.google.com/images/branding/googlelogo/1x/googlelogo_color_272x92dp.png'
}])[0]
def test_multiargs_autobatch():
# Make the maximum function
model = PythonStaticMethodModel.from_function_pointer(max, autobatch=True) \
.set_name('test').set_title('test')
# Describe the inputs
model.set_inputs('list', 'List of pairs of numbers',
item_type=compose_argument_block(
'tuple', 'Two numbers',
element_types=[
compose_argument_block('float', 'A number'),
compose_argument_block('float', 'A second number')
]))
model.set_outputs('list', 'Maximum of each pair', item_type='float')
model.set_unpack_inputs(True)
# Make sure the shim works
servable = PythonStaticMethodServable(**model.to_dict())
out, _ = servable.run([(1, 2)])
assert out == [2]
def test_file_multiinput():
model = PythonStaticMethodModel.from_function_pointer(multifile_input)
model.set_name('test')
model.set_title('test')
model.set_inputs('tuple', 'Several things', element_types=[
compose_argument_block('file', 'Single file'),
compose_argument_block('list', 'Multiple files', item_type='file'),
compose_argument_block('boolean', 'Something random')
])
model.set_outputs('bool', 'Should be True')
model.set_unpack_inputs(True)
# Make the servable
servable = PythonStaticMethodServable(**model.to_dict())
# Test it
assert servable.run([
{'url': __file__},
[{'url': __file__}],
True
])[0]