code: def testNonRectangularInputs(self):
      elements = [[[1]], [[2, 3]], [[4, 5, 6]]]
      dataset = from_list.from_list(elements)
      self.assertEqual(
          tensor_shape.Dimension(1),
          dataset_ops.get_legacy_output_shapes(dataset)[0],
      )
      self.assertDatasetProduces(dataset, expected_output=elements)
  
imports: import collections
  from absl.testing import parameterized
  import numpy
  from tensorflow.python.data.experimental.ops import from_list
  from tensorflow.python.data.experimental.ops import random_access
  from tensorflow.python.data.kernel_tests import checkpoint_test_base
  from tensorflow.python.data.kernel_tests import test_base
  from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.data.ops import options
  from tensorflow.python.framework import combinations
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import tensor_shape
  from tensorflow.python.platform import test
label_desc: Use image node_id 5 for calling the FromListTest obj's underlying member method code with example usage: obj.testNonRectangularInputs() without return types