code: def test_create_input(
      self,
      input_tensor_shape,
      feature_columns,
      problem_type,
      output_shape,
      input_tensor_shape2=None,
      lengths_shape=None,
      scope="Phoenix/Input",
  ):
      # Force graph mode
      with tf.compat.v1.Graph().as_default():
          features = {
              "feature": tf.zeros(input_tensor_shape),
          }
          lengths_feature_name = ""
          if lengths_shape:
              features["lengths"] = tf.ones(lengths_shape)
              lengths_feature_name = "lengths"
          if input_tensor_shape2:
              features["feature2"] = tf.ones(input_tensor_shape2)
  
          input_layer_fn = utils.default_get_input_layer_fn(
              problem_type, feature_columns
          )
  
          input_tensor, lengths = input_layer_fn(
              features,
              is_training=True,
              scope_name=scope,
              lengths_feature_name=lengths_feature_name,
          )
  
          self.assertAllEqual(input_tensor.shape, output_shape)
          self.assertIn(scope, input_tensor.name)
          if lengths_shape:
              self.assertAllEqual(lengths.shape, lengths_shape)
  
file_path: model_search/model_search/data/utils_test.py
imports: from absl.testing import parameterized
  from model_search.data import utils
  from model_search.proto import phoenix_spec_pb2
  import tensorflow.compat.v2
label_desc: Use image node_id 1 for calling the UtilsTest obj's underlying member method code with example usage: obj.test_create_input(input_tensor_shape, feature_columns, problem_type, output_shape, input_tensor_shape2, lengths_shape, scope) without return types