code: def test_sample():
      row = torch.tensor([0, 0, 2, 2])
      col = torch.tensor([1, 2, 0, 1])
      adj = SparseTensor(row=row, col=col, sparse_sizes=(3, 3))
  
      out = sample(adj, num_neighbors=1)
      assert out.min() >= 0 and out.max() <= 2
  
label_desc: Use image node_id 1 for calling a global function with example usage: test_sample() without return types