add test for deserialization (needs fixing)

This commit is contained in:
Philipp Moritz committed 2017-09-02 15:17:57 -07:00
1 parent 57772124c6
commit caad76d1d4
1 file changed
+16 -4
+16 -4
View File
@@ -256,15 +256,27 @@ def test_custom_serialization(large_memory_map):
for obj in CUSTOM_OBJECTS:
serialization_roundtrip(obj, mmap)
def test_serializaton_callback_error():
def test_serialization_callback_error():
class TempClass(object):
pass
# Have a SerializationContext but TempClass is not
# registered
# Pass a SerializationContext into serialize, but TempClass
# is not registered
serialization_context = pa.SerializationContext()
with pytest.raises(pa.SerializationCallbackError):
serialized_object = pa.serialize(TempClass, serialization_context)
serialized_object = pa.serialize(TempClass(), serialization_context)
serialization_context.register_type(TempClass, 20*b"\x00")
serialized_object = pa.serialize(TempClass(), serialization_context)
deserialization_context = pa.SerializationContext()
# Pass a Serialization Context into deserialize, but TempClass
# is not registered
with pytest.raises(pa.DeserializationCallbackError):
serialized_object.deserialize(deserialization_context)