code: def setUp(self):
      # Mock BigQuery result schema.
      self._schema = [
          bigquery.SchemaField(
              "context_feature_1", "INTEGER", mode="NULLABLE"
          ),
          bigquery.SchemaField(
              "context_feature_2", "INTEGER", mode="NULLABLE"
          ),
          bigquery.SchemaField(
              "feature_id_1", "INTEGER", mode="NULLABLE"
          ),
          bigquery.SchemaField(
              "feature_id_2", "FLOAT", mode="NULLABLE"
          ),
          bigquery.SchemaField(
              "feature_id_3", "STRING", mode="NULLABLE"
          ),
      ]
      super().setUp()
  
file_path: tfx/tfx/extensions/google_cloud_big_query/experimental/elwc_example_gen/component/executor_test.py
imports: from unittest import mock
  import apache_beam
  from apache_beam.testing import util
  from google.cloud import bigquery
  import tensorflow
  from tfx.extensions.google_cloud_big_query import utils
  from tfx.extensions.google_cloud_big_query.experimental.elwc_example_gen.component import executor
  from tfx.extensions.google_cloud_big_query.experimental.elwc_example_gen.proto import elwc_config_pb2
  from tfx.proto import example_gen_pb2
  from google.protobuf import json_format
  from google.protobuf import text_format
  from tensorflow_serving.apis import input_pb2
label_desc: Use image node_id 1 for calling the ExecutorTest obj's underlying member method code with example usage: obj.setUp() without return types