Skip to content

Commit d6fc48a

Browse files
committed
Add test for pickle serialization
1 parent 9386bbf commit d6fc48a

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

tests/unit/test_simvue.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -134,3 +134,14 @@ def test_pytorch_tensor_serialization():
134134
array_out = Deserializer().deserialize(serialized, mime_type)
135135

136136
assert (array == array_out).all()
137+
138+
def test_pickle_serialization():
139+
"""
140+
Check that a dictionary can be serialized then deserialized successfully
141+
"""
142+
data = {'a': 1.0, 'b': 'test'}
143+
144+
serialized, mime_type = Serializer().serialize(data, allow_pickle=True)
145+
data_out = Deserializer().deserialize(serialized, mime_type, allow_pickle=True)
146+
147+
assert (data == data_out)

0 commit comments

Comments
 (0)