code: def test_topk():
      raw = np.random.rand(20)
      a = tensor(raw, chunk_size=10)
  
      t = topk(a, 2)
      t = tile(t)
      assert t.op.parallel_kind == "tree"
  
      t = topk(a, 3)
      t = tile(t)
      assert t.op.parallel_kind == "psrs"
  
      t = topk(sort(a), 3)
      t = tile(t)
      # k is less than 100
      assert t.op.parallel_kind == "tree"
  
      with pytest.raises(ValueError):
          topk(a, 3, parallel_kind="unknown")
  
imports: import numpy
  import pytest
  from ....core import tile
  from ....core.operand import OperandStage
  from ...datasource import ones, tensor, arange
  from ..None import transpose, broadcast_to, where, argwhere, array_split, split, squeeze, result_type, repeat, copyto, isin, moveaxis, TensorCopyTo, atleast_1d, atleast_2d, atleast_3d, ravel, searchsorted, unique, sort, partition, topk, to_gpu, to_cpu
label_desc: Use image node_id 24 for calling a global function with example usage: test_topk() without return types