def test_default_output_fn(self, mxc):
import mxnet
from mxnet_container.serve.transformer import GluonBlockTransformer
mock_ndarray = mxnet.nd.array()
output, accept = GluonBlockTransformer._default_output_fn(
mock_ndarray, JSON_CONTENT_TYPE)
assert accept == JSON_CONTENT_TYPE
assert output == '[1, 2, 3]'
def test_default_predict_fn(self, mxc):
from mxnet_container.serve.transformer import GluonBlockTransformer
# block, ndarray could be any compatible callable/arg pair
block = list
ndarray = [1, 2, 3]
result = GluonBlockTransformer._default_predict_fn(block, ndarray)
assert [1, 2, 3] == result
def test_default_output_fn_unsupported_content_type(self, mxc):
from mxnet_container.serve.transformer import GluonBlockTransformer
with pytest.raises(UnsupportedAcceptTypeError):
GluonBlockTransformer._default_output_fn('whatever',
'wrong content type')
def test_default_input_fn(self, mxc):
import mxnet
from mxnet_container.serve.transformer import GluonBlockTransformer
_ = GluonBlockTransformer._default_input_fn('[[1,2,3,4]]',
JSON_CONTENT_TYPE)
mxnet.nd.array.assert_called_with([[1, 2, 3, 4]])