code: def test_merges_two_shards(self):
      stats1 = text_format.Parse(
          """
        datasets {
          name: 'x'
          num_examples: 100
          features: {
             path: {
                step: "f1"
             }
          }
        }
        """,
          statistics_pb2.DatasetFeatureStatisticsList(),
      )
      stats2 = text_format.Parse(
          """
          datasets {
            name: 'x'
            num_examples: 100
            features: {
               path: {
                  step: "f2"
               }
            }
          }
          """,
          statistics_pb2.DatasetFeatureStatisticsList(),
      )
  
      stats_combined = text_format.Parse(
          """
          datasets {
            name: 'x'
            num_examples: 100
            features: {
               path: {
                  step: "f1"
               }
            }
            features: {
               path: {
                  step: "f2"
               }
            }
          }
          """,
          statistics_pb2.DatasetFeatureStatisticsList(),
      )
      with beam.Pipeline() as p:
          result = (
              p
              | beam.Create([stats1, stats2])
              | stats_api.MergeDatasetFeatureStatisticsList()
          )
          util.assert_that(
              result,
              test_util.make_dataset_feature_stats_list_proto_equal_fn(
                  self, stats_combined, check_histograms=False
              ),
          )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  import tempfile
  from absl.testing import absltest
  import apache_beam
  from apache_beam.testing import util
  import numpy
  import pyarrow
  import tensorflow
  from tensorflow_data_validation.api import stats_api
  from tensorflow_data_validation.utils import artifacts_io_impl
  from tensorflow_data_validation.statistics import stats_options
  from tensorflow_data_validation.utils import io_util
  from tensorflow_data_validation.utils import stats_util
  from tensorflow_data_validation.utils import test_util
  from google.protobuf import text_format
  from tensorflow_metadata.proto.v0 import statistics_pb2
label_desc: Use image node_id 1 for calling the MergeDatasetFeatureStatisticsListTest obj's underlying member method code with example usage: obj.test_merges_two_shards() without return types