def test_csv_serializer_csv_reader():
csv_file_path = os.path.join(DATA_DIR, "with_integers.csv")
with open(csv_file_path) as csv_file:
validation_data = csv_file.read()
csv_file.seek(0)
result = csv_serializer(csv_file)
assert result == validation_data
def test_csv_serializer_csv_reader():
csv_file_path = os.path.join(DATA_DIR, "with_integers.csv")
with open(csv_file_path) as csv_file:
validation_data = csv_file.read()
csv_file.seek(0)
result = csv_serializer(csv_file)
assert result == validation_data
def __call__(self, data):
"""
Args:
data:
"""
to_serialize = data
if isinstance(data, tensor_pb2.TensorProto):
to_serialize = tensor_util.MakeNdarray(data)
return csv_serializer(to_serialize)
def __call__(self, data):
"""
Args:
data:
"""
to_serialize = data
from tensorflow.core.framework import tensor_pb2 # pylint: disable=no-name-in-module
from tensorflow.python.framework import tensor_util # pylint: disable=no-name-in-module
if isinstance(data, tensor_pb2.TensorProto):
to_serialize = tensor_util.MakeNdarray(data)
return csv_serializer(to_serialize)
def predict(payload):
global sm
global endpoint_name
global env
payload = payload
X = payload[1:]
y = payload[0]
elapsed_time = time.time()
resp = sm.invoke_endpoint(EndpointName=endpoint_name_mask % env,
ContentType='text/csv',
Accept='text/csv',
Body=csv_serializer(X))
elapsed_time = time.time() - elapsed_time
resp = float(resp['Body'].read().decode('utf-8').strip())
return (resp == y, elapsed_time)
def test_csv_serializer_python_invalid_empty():
with pytest.raises(ValueError) as error:
csv_serializer([])
assert "empty array" in str(error)
def test_csv_serializer_numpy_invalid_empty():
with pytest.raises(ValueError) as invalid_input:
csv_serializer(np.array([]))
assert "empty array" in str(invalid_input)
def test_csv_serializer_list_of_list():
result = csv_serializer([[1, 2, 3], [3, 4, 5]])
assert result == '1,2,3\n3,4,5'
def test_csv_serializer_list_of_str():
result = csv_serializer(['1,2,3', '4,5,6'])
assert result == '1,2,3\n4,5,6'
def test_csv_serializer_numpy_valid_2dimensional():
result = csv_serializer(np.array([[1, 2, 3], [3, 4, 5]]))
assert result == '1,2,3\n3,4,5'
def test_csv_serializer_numpy_invalid_empty():
with pytest.raises(ValueError) as invalid_input:
csv_serializer(np.array([]))
assert "empty array" in str(invalid_input)
def test_csv_serializer_python_array():
result = csv_serializer([1, 2, 3])
assert result == '1,2,3'
def test_csv_serializer_python_array():
result = csv_serializer([1, 2, 3])
assert result == '1,2,3'
def test_csv_serializer_numpy_valid():
result = csv_serializer(np.array([1, 2, 3]))
assert result == '1,2,3'
def test_csv_serializer_str():
original = '1,2,3'
result = csv_serializer('1,2,3')
assert result == original
def test_csv_serializer_list_of_str():
result = csv_serializer(['1,2,3', '4,5,6'])
assert result == '1,2,3\n4,5,6'
def test_csv_serializer_python_invalid_empty():
with pytest.raises(ValueError) as error:
csv_serializer([])
assert "empty array" in str(error)
def test_csv_serializer_numpy_valid_2dimensional():
result = csv_serializer(np.array([[1, 2, 3], [3, 4, 5]]))
assert result == '1,2,3\n3,4,5'
def test_csv_serializer_list_of_str():
result = csv_serializer(["1,2,3", "4,5,6"])
assert result == "1,2,3\n4,5,6"
def test_csv_serializer_str():
original = '1,2,3'
result = csv_serializer('1,2,3')
assert result == original
def test_csv_serializer_str():
original = "1,2,3"
result = csv_serializer("1,2,3")
assert result == original
def test_csv_serializer_numpy_valid():
result = csv_serializer(np.array([1, 2, 3]))
assert result == '1,2,3'
def test_csv_serializer_list_of_list():
result = csv_serializer([[1, 2, 3], [3, 4, 5]])
assert result == '1,2,3\n3,4,5'
