def get_path_value(self, start_id, steps=5): path = ttypes.Path() path.src = self.get_vertex_value(start_id) path.steps = list() for i in range(0, steps): step = ttypes.Step() step.dst = self.get_vertex_value('vertex{}'.format(i)) step.type = 1 if i % 2 == 0 else -1 step.name = 'classmate' step.ranking = 100 step.props = dict() for i in range(0, 5): step.props['prop{}'.format(i)] = i path.steps.append(step) return path
def get_path_value(self, start_id, steps=5): path = ttypes.Path() path.src = self.get_vertex_value(start_id) path.steps = list() for i in range(0, steps): step = ttypes.Step() step.dst = self.get_vertex_value(('vertex{}'.format(i)).encode('utf-8')) step.type = 1 if i % 2 == 0 else -1 step.name = b'classmate' step.ranking = 100 step.props = dict() for i in range(0, 5): value = ttypes.Value() value.set_iVal(i) step.props[('prop{}'.format(i)).encode('utf-8')] = value path.steps.append(step) return path
def to_value(self): path = CommonTtypes.Path() path.steps = [] for j, col in enumerate(self.items): if j == 0: path.src = col.get_vVal() elif (j % 2) == 1: edge = col[0].get_eVal() step = CommonTtypes.Step() step.name = edge.name step.ranking = edge.ranking step.type = col[1] step.props = edge.props path.steps.append(step) else: print("step: %d", len(path.steps)) path.steps[-1].dst = col.get_vVal() return path
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