TFL: Make the test more robust.

ANSI C requires len(argv) + 1 == argc and argv[argc] = 0

PiperOrigin-RevId: 256306649
This commit is contained in:
Tiezhen WANG 2019-07-02 23:18:34 -07:00 committed by TensorFlower Gardener
parent e0ab930a7e
commit 8719744db9

View File

@ -28,13 +28,12 @@ namespace {
TEST(ModelCmdlineFlagsTest, ParseArgsStringMapList) {
int args_count = 3;
const char* args[] = {
"toco",
"--input_arrays=input_1",
"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;