def test_input_fn_npz(np_array): input_data = encoders.array_to_npy(np_array) deserialized_np_array = serving.default_input_fn(input_data, content_types.NPY) assert np.array_equal(np_array, deserialized_np_array) float_32_array = np.array(np_array, dtype=np.float32) input_data = encoders.array_to_npy(float_32_array) deserialized_np_array = serving.default_input_fn(input_data, content_types.NPY) assert np.array_equal(float_32_array, deserialized_np_array) float_64_array = np.array(np_array, dtype=np.float64) input_data = encoders.array_to_npy(float_64_array) deserialized_np_array = serving.default_input_fn(input_data, content_types.NPY) assert np.array_equal(float_64_array, deserialized_np_array)
def test_output_fn_npz(np_array): response = serving.default_output_fn(np_array, content_types.NPY) assert response.get_data() == encoders.array_to_npy(np_array) assert response.content_type == content_types.NPY
def test_output_fn_npz(np_array): response = handler.default_output_fn(np_array, content_types.NPY) assert response == encoders.array_to_npy(np_array)