Skip to content

Commit 7a79f55

Browse files
author
Balaji Veeramani
committed
Add IdentitySerializer unit test
1 parent a805378 commit 7a79f55

File tree

2 files changed

+11
-1
lines changed

2 files changed

+11
-1
lines changed

src/sagemaker/serializers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -228,4 +228,4 @@ def serialize(self, data):
228228
"""
229229
buffer = io.BytesIO()
230230
scipy.sparse.save_npz(buffer, data)
231-
return buffer.getvalue()
231+
return buffer.getvalue()

tests/unit/sagemaker/test_serializers.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
CSVSerializer,
2525
NumpySerializer,
2626
JSONSerializer,
27+
IdentitySerializer,
2728
SparseMatrixSerializer,
2829
)
2930
from tests.unit import DATA_DIR
@@ -235,6 +236,15 @@ def test_json_serializer_csv_buffer(json_serializer):
235236
assert result == validation_value
236237

237238

239+
@pytest.fixture
240+
def identity_serializer():
241+
return IdentitySerializer()
242+
243+
244+
def test_identity_serializer(identity_serializer):
245+
assert identity_serializer.serialize(b"{}") == b"{}"
246+
247+
238248
@pytest.fixture
239249
def sparse_matrix_serializer():
240250
return SparseMatrixSerializer()

0 commit comments

Comments
 (0)