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:
parent
e0ab930a7e
commit
8719744db9
@ -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;
|
||||
|
Loading…
x
Reference in New Issue
Block a user