Function TEST

tensorflow/lite/toco/model_cmdline_flags_test.cc:28–56

namespace {

TEST(ModelCmdlineFlagsTest, ParseArgsStringMapList) {
  int args_count = 3;
  const char* args[] = {
      "toco", "--input_arrays=input_1",
      "--rnn_states={state_array:rnn/BasicLSTMCellZeroState/zeros,"
      "back_edge_source_array:rnn/basic_lstm_cell/Add_1,size:4},"
      "{state_array:rnn/BasicLSTMCellZeroState/zeros_1,"
      "back_edge_source_array:rnn/basic_lstm_cell/Mul_2,size:4}",
      nullptr};

  string expected_input_arrays = "input_1";
  std::vector<std::unordered_map<string, string>> expected_rnn_states;
  expected_rnn_states.push_back(
      {{"state_array", "rnn/BasicLSTMCellZeroState/zeros"},
       {"back_edge_source_array", "rnn/basic_lstm_cell/Add_1"},
       {"size", "4"}});
  expected_rnn_states.push_back(
      {{"state_array", "rnn/BasicLSTMCellZeroState/zeros_1"},
       {"back_edge_source_array", "rnn/basic_lstm_cell/Mul_2"},
       {"size", "4"}});

  string message;
  ParsedModelFlags result_flags;

  EXPECT_TRUE(ParseModelFlagsFromCommandLineFlags(
      &args_count, const_cast<char**>(args), &message, &result_flags));
  EXPECT_EQ(result_flags.input_arrays.value(), expected_input_arrays);
  EXPECT_EQ(result_flags.rnn_states.value().elements, expected_rnn_states);
}

}  // namespace
}  // namespace toco