code: def test_fps():
      N = 1000
      batch_size = 5
      sample_points = 10
      x = th.tensor(
          np.random.uniform(size=(batch_size, int(N / batch_size), 3))
      )
      ctx = F.ctx()
      if F.gpu_ctx():
          x = x.to(ctx)
      res = farthest_point_sampler(x, sample_points)
      assert res.shape[0] == batch_size
      assert res.shape[1] == sample_points
      assert res.sum() > 0
  
file_path: dgl/tests/python/pytorch/geometry/test_geometry.py
imports: import backend
  import dgl
  import dgl.nn
  import numpy
  import pytest
  import torch
  from dgl import DGLError
  from dgl.base import DGLWarning
  from dgl.geometry import farthest_point_sampler, neighbor_matching
  from utils import parametrize_idtype
  from utils.graph_cases import get_cases
label_desc: Use image node_id 1 for calling a global function with example usage: test_fps() without return types