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