def test_symmetric_json_conversion():
request_data = np.array([[5, 6, 7]])
datadef = scu.array_to_rest_datadef("ndarray", request_data)
json_request = {"jsonData": datadef}
seldon_message_request = scu.json_to_seldon_message(json_request)
result_json_request = scu.seldon_message_to_json(seldon_message_request)
assert json_request == result_json_request
def test_create_rest_response_jsondata_with_array_input():
user_model = UserObject(ret_dict=True)
request_data = np.array([[5, 6, 7]])
datadef = scu.array_to_rest_datadef("ndarray", request_data)
json_request = {"data": datadef}
raw_response = {"output": "data"}
json_response = scu.construct_response_json(user_model, True, json_request,
raw_response)
assert "data" not in json_response
assert json_response["jsonData"] == user_model.dict
def test_create_rest_response_jsondata():
user_model = UserObject()
request_data = np.array([[5, 6, 7]])
datadef = scu.array_to_rest_datadef("ndarray", request_data)
json_request = { "jsonData": datadef }
raw_response = {"output": "data"}
json_response = scu.construct_response_json(user_model, True, json_request, raw_response)
assert "data" not in json_response
emptyValue = Value()
assert json_response["jsonData"] != emptyValue
