code: def testMapAndBatchFusion(self):
      dataset = (
          dataset_ops.Dataset.range(10)
          .apply(testing.assert_next(["MapAndBatch"]))
          .map(lambda x: x * x)
          .batch(10)
      )
      options = options_lib.Options()
      options.experimental_optimization.apply_default_optimizations = (
          False
      )
      options.experimental_optimization.map_and_batch_fusion = True
      dataset = dataset.with_options(options)
      self.assertDatasetProduces(
          dataset, expected_output=[[x * x for x in range(10)]]
      )
  
file_path: tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/map_and_batch_fusion_test.py
imports: from absl.testing import parameterized
  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.framework import combinations
  from tensorflow.python.platform import test
label_desc: Use image node_id 1 for calling the MapAndBatchFusionTest obj's underlying member method code with example usage: obj.testMapAndBatchFusion() without return types