From c57034e631c2a3fa3db5fb6384e6dbec11a3d36d Mon Sep 17 00:00:00 2001 From: Jason Paryani Date: Thu, 17 Apr 2014 21:21:06 -0700 Subject: [PATCH] Add pickling test --- test/test_serialization.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/test/test_serialization.py b/test/test_serialization.py index 4c6203f..3928c31 100644 --- a/test/test_serialization.py +++ b/test/test_serialization.py @@ -4,6 +4,7 @@ import os import platform import test_regression import tempfile +import pickle this_dir = os.path.dirname(__file__) @@ -98,3 +99,11 @@ def test_file_and_bytes_packed(all_types): f.seek(0) assert f.read() == msg.to_bytes_packed() + +def test_pickle(all_types): + msg = all_types.TestAllTypes.new_message() + test_regression.init_all_types(msg) + data = pickle.dumps(msg) + msg2 = pickle.loads(data) + + test_regression.check_all_types(msg2)