code: def test_segment_top_k():
      x = np.array([0.2, 0.5, 0.3, -0.1, -0.2, -0.1], dtype=np.float32)
      I = np.array([0, 0, 0, 0, 1, 1], dtype=np.int64)
      ratio = 0.5
      topk = ops.segment_top_k(x, I, ratio)
      actual = topk.numpy()
      expected = [1, 2, 5]
      np.testing.assert_equal(actual, expected)
  
imports: import numpy
  import scipy.sparse
  import tensorflow
  from spektral.data.utils import to_disjoint
  from spektral.layers import ops
  from spektral.utils import convolution
  from spektral.utils.sparse import sp_batch_to_sp_tensor, sp_matrix_to_sp_tensor
label_desc: Use image node_id 15 for calling a global function with example usage: test_segment_top_k() without return types