code: def testConstructWithParameter(self):
      statistics_artifact = standard_artifacts.ExampleStatistics()
      statistics_artifact.split_names = (
          artifact_utils.encode_split_names(["train"])
      )
      infer_shape = data_types.RuntimeParameter(
          name="infer-shape", ptype=int
      )
      schema_gen = component.SchemaGen(
          statistics=channel_utils.as_channel([statistics_artifact]),
          infer_feature_shape=infer_shape,
      )
      self.assertEqual(
          standard_artifacts.Schema.TYPE_NAME,
          schema_gen.outputs[
              standard_component_specs.SCHEMA_KEY
          ].type_name,
      )
      self.assertJsonEqual(
          str(
              schema_gen.spec.exec_properties[
                  standard_component_specs.INFER_FEATURE_SHAPE_KEY
              ]
          ),
          str(infer_shape),
      )
  
file_path: tfx/tfx/components/schema_gen/component_test.py
imports: import tensorflow
  from tfx.components.schema_gen import component
  from tfx.orchestration import data_types
  from tfx.types import artifact_utils
  from tfx.types import channel_utils
  from tfx.types import standard_artifacts
  from tfx.types import standard_component_specs
label_desc: Use image node_id 2 for calling the SchemaGenTest obj's underlying member method code with example usage: obj.testConstructWithParameter() without return types