code: def test_jax_duck_typing_jax_array_methods(self):
      API_LIST = [
          "all",
          "any",
          "argmax",
          "argmin",
          "argsort",
          "conj",
          "conjugate",
          "cumprod",
          "cumsum",
          "diagonal",
          "flatten",
          "max",
          "mean",
          "min",
          "nonzero",
          "prod",
          "ptp",
          "ravel",
          "round",
          "sort",
          "squeeze",
          "std",
          "sum",
          "trace",
          "transpose",
          "var",
      ]
      v = objax.TrainVar(
          jn.array(
              [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0], [7.0, 8.0, 9.0]],
              dtype=jn.float32,
          )
      )
      for name in API_LIST:
          expected = getattr(v.value, name)()
          actual = getattr(v, name)()
          np.testing.assert_allclose(expected, actual)
      # other methods
      np.testing.assert_allclose(v.value.T, v.T)
      np.testing.assert_allclose(v.value.real, v.real)
      np.testing.assert_allclose(v.value.imag, v.imag)
      np.testing.assert_allclose(
          v.value.clip(0.0, 4.0), v.clip(0.0, 4.0)
      )
      np.testing.assert_allclose(v.value.dot(v.value), v.dot(v))
      np.testing.assert_allclose(v.value.repeat(3), v.repeat(3))
      np.testing.assert_allclose(
          v.value.swapaxes(0, 1), v.swapaxes(0, 1)
      )
      np.testing.assert_allclose(
          v.value.take(np.array([1, 2, 3])), v.take(np.array([1, 2, 3]))
      )
      np.testing.assert_allclose(
          jn.tile(v.value, [1, 2]), jn.tile(v, [1, 2])
      )
      np.testing.assert_allclose(v.value.reshape([-1]), v.reshape([-1]))
  
imports: import unittest
  import numpy
  import jax.numpy
  import objax
label_desc: Use image node_id 15 for calling the TestVariable obj's underlying member method code with example usage: obj.test_jax_duck_typing_jax_array_methods() without return types