code: def test_write_read(self):
      ta = nptf.compat.v2.TensorArray(
          dtype=np.float32,
          tensor_array_name="foo",
          infer_shape=False,
          dynamic_size=True,
      )
      w0 = ta.write(0, [[4.0, 5.0]])
      w1 = w0.write(1, [[1.0]])
      w2 = w1.write(2, -3.0)
      r0 = w2.read(0)
      r1 = w2.read(1)
      r2 = w2.read(2)
      self.assertAllEqual([[4.0, 5.0]], r0)
      self.assertAllEqual([[1.0]], r1)
      self.assertAllEqual(-3.0, r2)
  
file_path: probability/tensorflow_probability/python/internal/backend/numpy/tensor_array_ops_test.py
imports: import numpy
  from tensorflow_probability.python.internal.backend import numpy
  from tensorflow_probability.substrates.numpy.internal import test_util
label_desc: Use image node_id 2 for calling the TensorArrayTest obj's underlying member method code with example usage: obj.test_write_read() without return types