def setUp(self):
self.pb_serialize = lambda *args, **kw: \
serializes(ProtobufSerializeProvider(), *args, **kw)
self.pb_deserialize = lambda *args, **kw: \
deserializes(ProtobufSerializeProvider(), *args, **kw)
self.json_serialize = lambda *args, **kw: \
serializes(JsonSerializeProvider(), *args, **kw)
self.json_deserialize = lambda *args, **kw: \
deserializes(JsonSerializeProvider(), *args, **kw)
def setUp(self):
from mars.serialize import serializes, deserializes, \
ProtobufSerializeProvider, JsonSerializeProvider
self.pb_serialize = lambda *args, **kw: \
serializes(ProtobufSerializeProvider(), *args, **kw)
self.pb_deserialize = lambda *args, **kw: \
deserializes(ProtobufSerializeProvider(), *args, **kw)
self.json_serialize = lambda *args, **kw: \
serializes(JsonSerializeProvider(), *args, **kw)
self.json_deserialize = lambda *args, **kw: \
deserializes(JsonSerializeProvider(), *args, **kw)
def testDeserializeCls(self):
data = MockKeyObject(_data=10)
op = MockLog(_input=data)
serials = serializes(self.p, [data, op])
_, op2 = deserializes(self.p, [MockKeyObject, Operand], serials)
self.assertIsInstance(op2, Log)
self.assertBaseEqual(op, op2)
def pickle_unpickle(self, provider, *objs):
clses = [type(obj) for obj in objs]
serials = serializes(provider, objs)
if provider == self.j:
serials = [json.loads(json.dumps(s), object_hook=OrderedDict) for s in serials]
res = deserializes(provider, clses, serials)
if len(objs) == 1:
return res[0]
return res
