def test_1hop(self):
""" Test case for sample 1 hop neighbor with strategy of edge_weight.
All the src_ids have neighbors.
"""
expand_factor = 6
ids = self._seed_node1_ids
nbr_s = self.g.neighbor_sampler(self._edge1_type,
expand_factor=expand_factor,
strategy="edge_weight")
nbrs = nbr_s.get(ids)
edges = nbrs.layer_edges(1)
nodes = nbrs.layer_nodes(1)
utils.check_fixed_edge_dst_ids(edges, dst_range=self._node2_range,
expected_src_ids=ids)
utils.check_edge_type(edges,
src_type=self._node1_type,
dst_type=self._node2_type,
edge_type=self._edge1_type)
utils.check_edge_shape(edges, ids.size * expand_factor)
utils.check_edge_attrs(edges)
utils.check_edge_labels(edges)
utils.check_equal(nodes.ids, edges.dst_ids)
utils.check_node_ids(nodes, self._node2_ids)
utils.check_node_type(nodes, node_type=self._node2_type)
utils.check_node_shape(nodes, ids.size * expand_factor)
utils.check_node_weights(nodes)
utils.check_node_labels(nodes)
def test_1hop_using_gsl(self):
""" Using gsl api.
"""
gl.set_eager_mode(True)
expand_factor = 6
ids = self._seed_node1_ids
nbrs = self.g.V(self._node1_type, feed=ids) \
.outE(self._edge1_type).sample(expand_factor).by("in_degree") \
.inV().emit()
edges = nbrs[1]
nodes = nbrs[2]
utils.check_fixed_edge_dst_ids(edges,
dst_range=self._node2_range,
expected_src_ids=ids)
utils.check_edge_type(edges,
src_type=self._node1_type,
dst_type=self._node2_type,
edge_type=self._edge1_type)
utils.check_edge_shape(edges, ids.size * expand_factor)
utils.check_edge_attrs(edges)
utils.check_edge_labels(edges)
utils.check_equal(nodes.ids, edges.dst_ids)
utils.check_node_ids(nodes, self._node2_ids)
utils.check_node_type(nodes, node_type=self._node2_type)
utils.check_node_shape(nodes, ids.size * expand_factor)
utils.check_node_weights(nodes)
utils.check_node_labels(nodes)
def test_1hop(self):
""" Sample one hop of neighbors.
"""
expand_factor = 6
ids = self._seed_node1_ids
nbr_s = self.g.neighbor_sampler(self._edge1_type,
expand_factor=expand_factor,
strategy="random")
nbrs = nbr_s.get(ids)
edges = nbrs.layer_edges(1)
nodes = nbrs.layer_nodes(1)
print(edges.src_nodes.get_out_degrees(self._edge1_type))
utils.check_fixed_edge_dst_ids(edges, dst_range=self._node2_range,
expected_src_ids=ids)
utils.check_edge_type(edges,
src_type=self._node1_type,
dst_type=self._node2_type,
edge_type=self._edge1_type)
utils.check_edge_shape(edges, ids.size * expand_factor)
utils.check_edge_attrs(edges)
utils.check_edge_labels(edges)
utils.check_equal(nodes.ids, edges.dst_ids)
utils.check_node_ids(nodes, self._node2_ids)
utils.check_node_type(nodes, node_type=self._node2_type)
utils.check_node_shape(nodes, ids.size * expand_factor)
utils.check_node_weights(nodes)
utils.check_node_labels(nodes)
def test_1hop_using_gsl(self):
""" Test case for sample 1 hop neighbor.
hetegerous graph with edge attrs, without edge weight.
"""
gl.set_eager_mode(True)
expand_factor = 6
ids = self._seed_node1_ids
nbrs = self.g.V(self._node1_type, feed=ids) \
.outE(self._edge1_type).sample(expand_factor).by("edge_weight") \
.inV().emit()
edges = nbrs[1]
nodes = nbrs[2]
utils.check_fixed_edge_dst_ids(edges, dst_range=self._node2_range,
expected_src_ids=ids)
utils.check_edge_type(edges,
src_type=self._node1_type,
dst_type=self._node2_type,
edge_type=self._edge1_type)
utils.check_edge_shape(edges, ids.size * expand_factor)
utils.check_edge_attrs(edges)
utils.check_edge_labels(edges)
utils.check_equal(nodes.ids, edges.dst_ids)
utils.check_node_ids(nodes, self._node2_ids)
utils.check_node_type(nodes, node_type=self._node2_type)
utils.check_node_shape(nodes, ids.size * expand_factor)
utils.check_node_weights(nodes)
utils.check_node_labels(nodes)
def test_2hop(self):
""" Sample 2 hops of neighbors.
"""
gl.set_eager_mode(True)
expand_factor = [3, 2]
ids = self._seed_node1_ids
nbr_s = self.g.neighbor_sampler([self._edge1_type, self._edge2_type],
expand_factor=expand_factor,
strategy="in_degree")
nbrs = nbr_s.get(ids)
edges = nbrs.layer_edges(1)
nodes = nbrs.layer_nodes(1)
utils.check_fixed_edge_dst_ids(edges,
dst_range=self._node2_range,
expected_src_ids=ids)
utils.check_edge_type(edges,
src_type=self._node1_type,
dst_type=self._node2_type,
edge_type=self._edge1_type)
utils.check_edge_shape(edges, ids.size * expand_factor[0])
utils.check_edge_attrs(edges)
utils.check_edge_labels(edges)
utils.check_equal(nodes.ids, edges.dst_ids)
utils.check_node_ids(nodes, self._node2_ids)
utils.check_node_type(nodes, node_type=self._node2_type)
utils.check_node_shape(nodes, ids.size * expand_factor[0])
utils.check_node_weights(nodes)
utils.check_node_labels(nodes)
ids = nodes.ids.reshape(-1)
edges = nbrs.layer_edges(2)
nodes = nbrs.layer_nodes(2)
utils.check_fixed_edge_dst_ids(edges,
dst_range=self._node1_range,
expected_src_ids=ids)
utils.check_edge_type(edges,
src_type=self._node2_type,
dst_type=self._node1_type,
edge_type=self._edge2_type)
utils.check_edge_shape(edges, ids.size * expand_factor[1])
utils.check_edge_attrs(edges)
utils.check_edge_weights(edges)
utils.check_equal(nodes.ids, edges.dst_ids)
utils.check_node_ids(nodes, self._node1_ids)
utils.check_node_type(nodes, node_type=self._node1_type)
utils.check_node_shape(nodes, ids.size * expand_factor[1])
def test_1hop_using_gremlin(self):
""" Full neighbor sample with gremlin-like api.
"""
expand_factor = 2
ids = self._seed_node1_ids
nbrs = self.g.V(self._node1_type, feed=ids) \
.outE(self._edge1_type).sample().by("full") \
.inV().emit()
nodes = nbrs[2]
index = 0
for node in nodes:
utils.check_sorted_equal(
utils.fixed_dst_ids(ids[index], self._node2_range), node.ids)
index += 1
utils.check_node_ids(nodes, self._node2_ids)
utils.check_node_type(nodes, node_type=self._node2_type)
utils.check_node_weights(nodes)
utils.check_node_labels(nodes)
