/* expected_dtype = */ DT_FLOAT))
      << "Expected " << flattened[0]->DebugString();
}

TEST(SignatureFlatteningTest, ThreeArgInputSignature) {
  std::vector<const TensorSpecProto*> flattened;
  StructuredValue value = testing::ThreeArgInputSignature();
  TF_EXPECT_OK(internal::FlattenSignature(value, &flattened));
  EXPECT_EQ(flattened.size(), 3);
  EXPECT_TRUE(TensorSpecsAreEqual(*flattened[0],

// tf.function with 1 input
StructuredValue SingleArgInputSignature();

// Returns a StructuredValue corresponding to the serialized InputSignature of a
// tf.function with 3 inputs
StructuredValue ThreeArgInputSignature();

// Returns a StructuredValue corresponding to the serialized OutputSignature of
// a tf.function with no return values
StructuredValue ZeroReturnOutputSignature();

TEST_F(SavedConcreteFunctionLoadingTest, TooManyInputsInSavedConcreteFunction) {
  // `saved` has 3 inputs
  SavedConcreteFunction saved;
  *saved.mutable_canonicalized_input_signature() =
      testing::ThreeArgInputSignature();
  *saved.mutable_output_signature() = testing::ZeroReturnOutputSignature();

  // `func` has 2 inputs
  FunctionDef func = FuncDefWithNumInputsOutputs(2, 0);

  std::unique_ptr<TFConcreteFunction> result;

  return ParseStructuredValue(kZeroArgInputSignatureTextProto);
}

StructuredValue SingleArgInputSignature() {
  return ParseStructuredValue(kSingleArgInputSignatureTextProto);
}

StructuredValue ThreeArgInputSignature() {
  return ParseStructuredValue(kThreeArgInputSignatureTextProto);
}

StructuredValue ZeroReturnOutputSignature() {
  return ParseStructuredValue(kZeroReturnOutputSignatureTextProto);
}