code: def test_rw_small_with_edge_indices(device):
      row = tensor([0, 1], torch.long, device)
      col = tensor([1, 0], torch.long, device)
      start = tensor([0, 1, 2], torch.long, device)
      walk_length = 4
  
      node_seq, edge_seq = random_walk(
          row,
          col,
          start,
          walk_length,
          num_nodes=3,
          return_edge_indices=True,
      )
      assert node_seq.tolist() == [
          [0, 1, 0, 1, 0],
          [1, 0, 1, 0, 1],
          [2, 2, 2, 2, 2],
      ]
      assert edge_seq.tolist() == [
          [0, 1, 0, 1],
          [1, 0, 1, 0],
          [-1, -1, -1, -1],
      ]
  
imports: import pytest
  import torch
  from torch_cluster import random_walk
  from torch_cluster.testing import devices, tensor
label_desc: Use image node_id 4 for calling a global function with example usage: test_rw_small_with_edge_indices(device) without return types