code: def test_vectorize_random_function(self):
      class RandomReverse(objax.Module):
          def __init__(self):
              self.keygen = objax.random.Generator()
  
          def __call__(self, x):
              r = objax.random.randint(
                  [], low=0, high=2, generator=self.keygen
              )
              return x + r * (x[::-1] - x), r
  
      random_reverse = RandomReverse()
      vector_reverse = objax.Vectorize(random_reverse)
      x = jn.arange(20).reshape((10, 2))
      y, r = vector_reverse(x)
      self.assertEqual(r.tolist(), [1, 1, 1, 1, 1, 0, 0, 1, 0, 0])
      self.assertEqual(
          y.tolist(),
          [
              [1, 0],
              [3, 2],
              [5, 4],
              [7, 6],
              [9, 8],
              [10, 11],
              [12, 13],
              [15, 14],
              [16, 17],
              [18, 19],
          ],
      )
  
file_path: objax/tests/vectorize.py
imports: import unittest
  import jax.numpy
  import numpy
  import objax
label_desc: Use image node_id 9 for calling the TestVectorize obj's underlying member method code with example usage: obj.test_vectorize_random_function() and returns: unknown,  r