code: def testOptionsGraphRoundTripOptimization(self, map_parallelization):
      dataset = dataset_ops.Dataset.range(6)
      options = options_lib.Options()
      options.experimental_optimization.map_parallelization = (
          map_parallelization
      )
      dataset = dataset.with_options(options)
      dataset = self.graphRoundTrip(dataset)
      expected = "ParallelMap" if map_parallelization else "Map"
      dataset = dataset.apply(testing.assert_next([expected]))
      dataset = dataset.map(lambda x: x * x)
      self.assertDatasetProduces(
          dataset, expected_output=[0, 1, 4, 9, 16, 25]
      )
  
imports: from absl.testing import parameterized
  from tensorflow.core.framework import dataset_options_pb2
  from tensorflow.python.data.experimental.ops import testing
  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.eager import context
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import combinations
  from tensorflow.python.platform import test
label_desc: Use image node_id 20 for calling the OptionsTest obj's underlying member method code with example usage: obj.testOptionsGraphRoundTripOptimization(map_parallelization) without return types