code: def test_unbatch_edge_index():
      edge_index = torch.tensor(
          [
              [0, 1, 1, 2, 2, 3, 4, 5, 5, 6],
              [1, 0, 2, 1, 3, 2, 5, 4, 6, 5],
          ]
      )
      batch = torch.tensor([0, 0, 0, 0, 1, 1, 1])
  
      edge_indices = unbatch_edge_index(edge_index, batch)
      assert edge_indices[0].tolist() == [
          [0, 1, 1, 2, 2, 3],
          [1, 0, 2, 1, 3, 2],
      ]
      assert edge_indices[1].tolist() == [[0, 1, 1, 2], [1, 0, 2, 1]]
  
label_desc: Use image node_id 2 for calling a global function with example usage: test_unbatch_edge_index() without return types