def get_vertex_value(self, vid):
vertex = ttypes.Vertex()
vertex.vid = vid
vertex.tags = list()
for i in range(0, 3):
tag = ttypes.Tag()
tag.name = 'tag{}'.format(i)
tag.props = dict()
for j in range(0, 5):
tag.props['prop{}'.format(j)] = j
vertex.tags.append(tag)
return vertex
def check_path_result_without_prop(self, rows, expect):
msg = 'len(rows)[%d] != len(expect)[%d]' % (len(rows), len(expect))
assert len(rows) == len(expect), msg
for exp in expect:
path = CommonTtypes.Path()
path.steps = []
for col, j in zip(exp, range(len(exp))):
if j == 0:
src = CommonTtypes.Vertex()
src.vid = bytes(col, encoding='utf-8')
src.tags = []
path.src = src
else:
assert len(col) == 3, \
"{} invalid values size in expect result".format(exp.__repr__())
step = CommonTtypes.Step()
step.name = bytes(col[0], encoding='utf-8')
step.ranking = col[1]
step.type = 1
dst = CommonTtypes.Vertex()
dst.vid = bytes(col[2], encoding='utf-8')
dst.tags = []
step.dst = dst
step.props = {}
path.steps.append(step)
find = False
for row in rows:
assert len(row.values) == 1, \
"invalid values size in rows: {}".format(row)
assert row.values[0].getType() == CommonTtypes.Value.PVAL, \
"invalid column path type: {}".format(row.values[0].getType()())
if row.values[0].get_pVal() == path:
find = True
break
msg = self.check_format_str.format(row.values[0].get_pVal(), path)
assert find, msg
rows.remove(row)
assert len(rows) == 0
def get_vertex_value(self, vid):
vertex = ttypes.Vertex()
vertex.vid = vid
vertex.tags = list()
for i in range(0, 3):
tag = ttypes.Tag()
tag.name = ('tag{}'.format(i)).encode('utf-8')
tag.props = dict()
for j in range(0, 5):
value = ttypes.Value()
value.set_iVal(j)
tag.props[('prop{}'.format(j)).encode('utf-8')] = value
vertex.tags.append(tag)
return vertex
def create_vertex_team(line):
assert len(line) == 2
vertex = ttypes.Vertex()
vertex.vid = utf8b(line[0])
tags = []
tag = ttypes.Tag()
tag.name = utf8b('team')
props = dict()
name = ttypes.Value()
name.set_sVal(utf8b(line[1]))
props[utf8b('name')] = name
tag.props = props
tags.append(tag)
vertex.tags = tags
return vertex
def create_vertex_team(self, line):
if len(line) != 2:
assert False
vertex = CommonTtypes.Vertex()
vertex.vid = bytes(line[0], encoding='utf-8')
tags = []
tag = CommonTtypes.Tag()
tag.name = bytes('team', encoding='utf-8')
props = dict()
name = CommonTtypes.Value()
name.set_sVal(bytes(line[1], encoding='utf-8'))
props[bytes('name', encoding='utf-8')] = name
tag.props = props
tags.append(tag)
vertex.tags = tags
return vertex
def create_vertex_bachelor(line):
assert len(line) == 3
vertex = ttypes.Vertex()
vertex.vid = utf8b(line[0])
tags = []
tag = ttypes.Tag()
tag.name = utf8b('bachelor')
props = dict()
name = ttypes.Value()
name.set_sVal(utf8b(line[1]))
props[utf8b('name')] = name
speciality = ttypes.Value()
speciality.set_sVal(utf8b(line[2]))
props[utf8b('speciality')] = speciality
tag.props = props
tags.append(tag)
vertex.tags = tags
return vertex
def create_vertex_player(line):
assert len(line) == 3
vertex = ttypes.Vertex()
vertex.vid = utf8b(line[0])
tags = []
tag = ttypes.Tag()
tag.name = utf8b('player')
props = dict()
name = ttypes.Value()
name.set_sVal(utf8b(line[1]))
props[utf8b('name')] = name
age = ttypes.Value()
age.set_iVal(int(line[2]))
props[utf8b('age')] = age
tag.props = props
tags.append(tag)
vertex.tags = tags
return vertex
def create_vertex_bachelor(self, line):
if len(line) != 3:
assert False
vertex = CommonTtypes.Vertex()
vertex.vid = bytes(line[0], encoding='utf-8')
tags = []
tag = CommonTtypes.Tag()
tag.name = bytes('bachelor', encoding='utf-8')
props = dict()
name = CommonTtypes.Value()
name.set_sVal(bytes(line[1], encoding='utf-8'))
props[bytes('name', encoding='utf-8')] = name
speciality = CommonTtypes.Value()
speciality.set_sVal(bytes(line[2], encoding='utf-8'))
props[bytes('speciality', encoding='utf-8')] = speciality
tag.props = props
tags.append(tag)
vertex.tags = tags
return vertex
def create_vertex_player(self, line):
if len(line) != 3:
assert False
vertex = CommonTtypes.Vertex()
vertex.vid = bytes(line[0], encoding='utf-8')
tags = []
tag = CommonTtypes.Tag()
tag.name = bytes('player', encoding='utf-8')
props = dict()
name = CommonTtypes.Value()
name.set_sVal(bytes(line[1], encoding='utf-8'))
props[bytes('name', encoding='utf-8')] = name
age = CommonTtypes.Value()
age.set_iVal(int(line[2]))
props[bytes('age', encoding='utf-8')] = age
tag.props = props
tags.append(tag)
vertex.tags = tags
return vertex
def gen_vertex(cls, vid, tags):
vertex = ttypes.Vertex()
vertex.vid = vid
vertex.tags = tags
return vertex
