Function TEST

tensorflow/contrib/session_bundle/bundle_shim_test.cc:115–128

Checks that the input map in a signature def is populated correctly.

// Checks that the input map in a signature def is populated correctly.
TEST(BundleShimTest, AddInputToSignatureDef) {
  SignatureDef signature_def;
  const string tensor_name = "foo_tensor";
  const string map_key = "foo_key";

  // Build a map of tensor-name to dtype, for the unit-test.
  std::unordered_map<string, DataType> tensor_name_to_dtype;
  tensor_name_to_dtype[tensor_name] = tensorflow::DT_STRING;

  AddInputToSignatureDef(tensor_name, tensor_name_to_dtype, map_key,
                         &signature_def);
  EXPECT_EQ(1, signature_def.inputs_size());
  EXPECT_EQ(tensor_name, signature_def.inputs().find(map_key)->second.name());
}

// Checks that the output map in a signature def is populated correctly.
TEST(BundleShimTest, AddOutputToSignatureDef) {