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(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): """ 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_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_labeled(self): file_path = self.gen_test_data([utils.LABELED], False) decoder = gl.Decoder(labeled=True) g = gl.Graph() \ .edge(source=file_path, edge_type=self.edge_tuple_, decoder=decoder) g.init(server_id=0, server_count=1, tracker=utils.TRACKER_PATH) edges = g.E("first").batch(self.batch_size_).emit() utils.check_ids(edges.src_ids, range(self.src_range_[0], self.src_range_[1])) utils.check_ids(edges.dst_ids, range(self.dst_range_[0], self.dst_range_[1])) utils.check_edge_labels(edges)
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_weighted_labeled(self): file_path = self.gen_test_data([utils.WEIGHTED, utils.LABELED], False) decoder = gl.Decoder(weighted=True, labeled=True) g = gl.Graph() \ .edge(source=file_path, edge_type=self.edge_tuple_, decoder=decoder) g.init(tracker=utils.TRACKER_PATH) edges = g.E("first").batch(self.batch_size_).emit() utils.check_ids(edges.src_ids, range(self.src_range_[0], self.src_range_[1])) utils.check_ids(edges.dst_ids, range(self.dst_range_[0], self.dst_range_[1])) utils.check_edge_labels(edges) utils.check_edge_weights(edges) g.close()
def test_labeled(self): file_path = self.gen_test_data([utils.LABELED], False) decoder = gl.Decoder(labeled=True) g = gl.Graph() \ .edge(source=file_path, edge_type=self.edge_tuple_, decoder=decoder) g.init(tracker=utils.TRACKER_PATH) query = g.E("first").batch(self.batch_size_).alias('e').values() ds = gl.Dataset(query, window=1) edges = ds.next()['e'] utils.check_ids(edges.src_ids, range(self.src_range_[0], self.src_range_[1])) utils.check_ids(edges.dst_ids, range(self.dst_range_[0], self.dst_range_[1])) utils.check_edge_labels(edges) g.close()
def test_labeled_attributed(self): gl.set_eager_mode(True) file_path = self.gen_test_data([utils.LABELED, utils.ATTRIBUTED], False) decoder = gl.Decoder(labeled=True, attr_types=utils.ATTR_TYPES) g = gl.Graph() \ .edge(source=file_path, edge_type=self.edge_tuple_, decoder=decoder) g.init(tracker=utils.TRACKER_PATH) edges = g.E("first").batch(self.batch_size_).emit() utils.check_ids(edges.src_ids, range(self.src_range_[0], self.src_range_[1])) utils.check_ids(edges.dst_ids, range(self.dst_range_[0], self.dst_range_[1])) utils.check_edge_labels(edges) utils.check_edge_attrs(edges) g.close()
def test_weighted_labeled_attributed(self): file_path = self.gen_test_data( [utils.WEIGHTED, utils.LABELED, utils.ATTRIBUTED], False) decoder = gl.Decoder( weighted=True, labeled=True, attr_types=utils.ATTR_TYPES) g = gl.Graph() \ .edge(source=file_path, edge_type=self.edge_tuple_, decoder=decoder) g.init(tracker=utils.TRACKER_PATH) query = g.E("first").batch(self.batch_size_).alias('e').values() ds = gl.Dataset(query) edges = ds.next()['e'] utils.check_ids(edges.src_ids, range(self.src_range_[0], self.src_range_[1])) utils.check_ids(edges.dst_ids, range(self.dst_range_[0], self.dst_range_[1])) utils.check_edge_labels(edges) utils.check_edge_attrs(edges) utils.check_edge_weights(edges) g.close()