code: def test_sample(self):
      X = Tensor.rand(10000, 50).realize()
      BS = 16
      idxs = np.random.randint(0, X.shape[0], size=(BS))
      # this uncovered a bug with arg sort order
      batch = [
          Variable(f"idx{i}", 0, X.shape[0] - 1).bind(s)
          for i, s in enumerate(idxs.tolist())
      ]
      x = Tensor.cat(
          *[
              X.shrink(((batch[i], batch[i] + 1), None))
              for i in range(BS)
          ]
      )
      print(idxs)
      ret = x.numpy()
      base = X.numpy()[idxs]
      np.testing.assert_equal(ret, base)
  
imports: import unittest
  import numpy
  from tinygrad.tensor import Tensor
  from tinygrad.shape.symbolic import Variable
label_desc: Use image node_id 1 for calling the TestSample obj's underlying member method code with example usage: obj.test_sample() without return types