code: def test_set_domain(
      self,
      input_schema_proto_text,
      feature_name_or_path,
      domain,
      output_schema_proto_text,
  ):
      actual_schema = schema_pb2.Schema()
      text_format.Merge(input_schema_proto_text, actual_schema)
      schema_util.set_domain(
          actual_schema, feature_name_or_path, domain
      )
      expected_schema = schema_pb2.Schema()
      text_format.Merge(output_schema_proto_text, expected_schema)
      self.assertEqual(actual_schema, expected_schema)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  from absl import flags
  from absl.testing import absltest
  from absl.testing import parameterized
  from tensorflow_data_validation import types
  from tensorflow_data_validation.utils import schema_util
  from google.protobuf import text_format
  from tensorflow_metadata.proto.v0 import schema_pb2
label_desc: Use image node_id 21 for calling the SchemaUtilTest obj's underlying member method code with example usage: obj.test_set_domain(input_schema_proto_text, feature_name_or_path, domain, output_schema_proto_text) without return types