code: def test_output_is_correct(self):
      tensor_splits = (("x1", 1), ("x2", 2), ("x3", 3))
      x1 = np.zeros((2, 3, 1), dtype=np.float32) + 1.0
      x2 = np.zeros((2, 3, 2), dtype=np.float32) + 2.0
      x3 = np.zeros((2, 3, 3), dtype=np.float32) + 3.0
      x = tf.constant(np.concatenate([x1, x2, x3], axis=2))
  
      output = nn.split_to_dict(x, tensor_splits)
  
      self.assertSetEqual(set(["x1", "x2", "x3"]), set(output.keys()))
      self.assertAllEqual(x1, output.get("x1"))
      self.assertAllEqual(x2, output.get("x2"))
      self.assertAllEqual(x3, output.get("x3"))
  
imports: from absl.testing import parameterized
  from ddsp.training import nn
  import numpy
  import tensorflow.compat.v2
label_desc: Use image node_id 1 for calling the SplitToDictTest obj's underlying member method code with example usage: obj.test_output_is_correct() without return types