code: def test_knn_interpolate():
      x = torch.tensor(
          [[1.0], [10.0], [100.0], [-1.0], [-10.0], [-100.0]]
      )
      pos_x = torch.tensor(
          [
              [-1.0, 0.0],
              [0.0, 0.0],
              [1.0, 0.0],
              [-2.0, 0.0],
              [0.0, 0.0],
              [2.0, 0.0],
          ]
      )
      pos_y = torch.tensor(
          [
              [-1.0, -1.0],
              [1.0, 1.0],
              [-2.0, -2.0],
              [2.0, 2.0],
          ]
      )
      batch_x = torch.tensor([0, 0, 0, 1, 1, 1])
      batch_y = torch.tensor([0, 0, 1, 1])
  
      y = knn_interpolate(x, pos_x, pos_y, batch_x, batch_y, k=2)
      assert y.tolist() == [[4.0], [70.0], [-4.0], [-70.0]]
  
label_desc: Use image node_id 1 for calling a global function with example usage: test_knn_interpolate() without return types