code: def setUp(self):
      self._unbatched_states = [
          (
              np.array([[1, 2, 3], [4, 5, 6]]),
              (np.array([7, 8]), np.array([9])),
              np.array([[10], [11]]),
          ),
          (
              np.array([[12, 13, 14], [15, 16, 17]]),
              (np.array([18, 19]), np.array([20])),
              np.array([[21], [22]]),
          ),
      ]
  
      self._batched_states = (
          np.array(
              [
                  [[1, 2, 3], [4, 5, 6]],
                  [[12, 13, 14], [15, 16, 17]],
                  [[0, 0, 0], [0, 0, 0]],
              ]
          ),
          (
              np.array([[7, 8], [18, 19], [0, 0]]),
              np.array([[9], [20], [0]]),
          ),
          np.array([[[10], [11]], [[21], [22]], [[0], [0]]]),
      )
  
file_path: magenta/magenta/common/state_util_test.py
imports: from magenta.common import state_util
  import numpy
  import tensorflow.compat.v1
label_desc: Use image node_id 1 for calling the StateUtilTest obj's underlying member method code with example usage: obj.setUp() without return types