code: def test_index_to_mask():
      index = torch.tensor([1, 3, 5])
  
      mask = index_to_mask(index)
      assert mask.tolist() == [False, True, False, True, False, True]
  
      mask = index_to_mask(index, size=7)
      assert mask.tolist() == [
          False,
          True,
          False,
          True,
          False,
          True,
          False,
      ]
  
imports: import torch
  from torch_geometric.utils import index_to_mask, mask_select, mask_to_index
label_desc: Use image node_id 2 for calling a global function with example usage: test_index_to_mask() without return types