code: def test_neighbor_sample_mock_sampling_results(abc_graph):
      F, G, N = abc_graph
  
      graph_store = CuGraphStore(F, G, N, order="CSR")
  
      # let 0, 1 be the start vertices, fanout = [2, 1, 2, 3]
      mock_sampling_results = cudf.DataFrame(
          {
              "majors": cudf.Series(
                  [0, 0, 1, 2, 3, 3, 1, 3, 3, 3], dtype="int64"
              ),
              "minors": cudf.Series(
                  [2, 3, 3, 8, 1, 7, 3, 1, 5, 7], dtype="int64"
              ),
              "hop_id": cudf.Series(
                  [0, 0, 0, 1, 1, 1, 2, 3, 3, 3], dtype="int32"
              ),
              "edge_type": cudf.Series(
                  [0, 0, 0, 2, 1, 2, 0, 1, 2, 2], dtype="int32"
              ),
          }
      )
  
      out = _sampler_output_from_sampling_results_heterogeneous(
          mock_sampling_results, None, graph_store, None
      )
  
      assert out.metadata is None
      assert len(out.node) == 3
      assert out.node["A"].tolist() == [0, 1]
      assert out.node["B"].tolist() == [0, 1]
      assert out.node["C"].tolist() == [3, 2, 0]
  
      assert len(out.row) == 3
      assert len(out.col) == 3
      assert out.row[("A", "ab", "B")].tolist() == [0, 0, 1, 1]
      assert out.col[("A", "ab", "B")].tolist() == [0, 1, 1, 1]
      assert out.row[("B", "bc", "C")].tolist() == [0, 1, 1, 1]
      assert out.col[("B", "bc", "C")].tolist() == [0, 1, 2, 1]
      assert out.row[("B", "ba", "A")].tolist() == [1, 1]
      assert out.col[("B", "ba", "A")].tolist() == [1, 1]
  
      assert len(out.num_sampled_nodes) == 3
      assert out.num_sampled_nodes["A"].tolist() == [2, 0, 0, 0, 0]
      assert out.num_sampled_nodes["B"].tolist() == [0, 2, 0, 0, 0]
      assert out.num_sampled_nodes["C"].tolist() == [0, 0, 2, 0, 1]
  
      assert len(out.num_sampled_edges) == 3
      assert out.num_sampled_edges[("A", "ab", "B")].tolist() == [
          3,
          0,
          1,
          0,
      ]
      assert out.num_sampled_edges[("B", "ba", "A")].tolist() == [
          0,
          1,
          0,
          1,
      ]
      assert out.num_sampled_edges[("B", "bc", "C")].tolist() == [
          0,
          2,
          0,
          2,
      ]
  
file_path: cugraph/python/cugraph-pyg/cugraph_pyg/tests/test_cugraph_sampler.py
imports: import cudf
  import cupy
  import pytest
  from cugraph_pyg.data import CuGraphStore
  from cugraph_pyg.sampler.cugraph_sampler import _sampler_output_from_sampling_results_heterogeneous
  from cugraph.utilities.utils import import_optional, MissingModule
  from cugraph import uniform_neighbor_sample
label_desc: Use image node_id 3 for calling a global function with example usage: test_neighbor_sample_mock_sampling_results(abc_graph) without return types