class Node4(AttributeAsDict):
attr_tag = 'attr'
a = BytesField('b')
b = NDArrayField('c')
c = Datetime64Field('d')
d = Timedelta64Field('e')
e = DataTypeField('f')
f = DictField('g', ValueType.string, ValueType.list(ValueType.bool))
g = DictField('h')
h = TupleField('i', ValueType.int64, ValueType.unicode, ValueType.string,
ValueType.float32, ValueType.datetime64,
ValueType.timedelta64, ValueType.dtype)
i = TupleField('j', ValueType.slice)
j = ReferenceField('k', Node5)
k = ListField('l', ValueType.reference('Node5'))
l = OneOfField('m', n5=Node5, n6=Node6)
m = IndexField('n')
mm = IndexField('mn')
n = SeriesField('o')
o = DataFrameField('p')
p = ListField('q')
@classmethod
def cls(cls, provider):
if provider.type == ProviderType.protobuf:
from mars.serialize.tests.testser_pb2 import Node4Def
return Node4Def
return super(Node4, cls).cls(provider)
class Node1(Serializable):
a = IdentityField('a', ValueType.string)
b1 = Int8Field('b1')
b2 = Int16Field('b2')
b3 = Int32Field('b3')
b4 = Int64Field('b4')
c1 = UInt8Field('c1')
c2 = UInt16Field('c2')
c3 = UInt32Field('c3')
c4 = UInt64Field('c4')
d1 = Float16Field('d1')
d2 = Float32Field('d2')
d3 = Float64Field('d3')
cl1 = Complex64Field('cl1')
cl2 = Complex128Field('cl2')
e = BoolField('e')
f = KeyField('f')
g = ReferenceField('g', 'Node2')
h = ListField('h')
i = ListField('i', ValueType.reference('self'))
j = ReferenceField('j', None)
k = ListField('k', ValueType.reference(None))
def __new__(cls, *args, **kwargs):
if 'a' in kwargs and kwargs['a'] == 'test1':
return object.__new__(Node8)
return object.__new__(cls)
@classmethod
def cls(cls, provider):
if provider.type == ProviderType.protobuf:
from mars.serialize.tests.testser_pb2 import Node1Def
return Node1Def
return super().cls(provider)
class Node6(AttributeAsDict):
nid = IdentityField('id', ValueType.int64)
b = Int32Field('b')
r = ReferenceField('r', 'self')
rl = ListField('rl', ValueType.reference('self'))
def __new__(cls, *args, **kwargs):
if 'nid' in kwargs and kwargs['nid'] != 0:
return object.__new__(Node7)
return object.__new__(cls)
class Node2(Serializable, BaseWithKey):
a = ListField('a', ValueType.list(ValueType.string))
_key = StringField('key')
_id = StringField('id')
_name = UnicodeField('name')
data = ListField('data', ValueType.int32)
@classmethod
def cls(cls, provider):
if provider.type == ProviderType.protobuf:
from mars.serialize.tests.testser_pb2 import Node2Def
return Node2Def
return super(Node2, cls).cls(provider)
