Function TestCase1

tensorflow/core/kernels/data/zip_dataset_op_test.cc:90–101

Test case 1: the input datasets with same number of outputs.

// Test case 1: the input datasets with same number of outputs.
TestParam TestCase1() {
  return {/*input_range_dataset_params*/
          {RangeDatasetParam{0, 3, 1}, RangeDatasetParam{10, 13, 1}},
          /*expected_outputs*/
          {CreateTensor<int64>(TensorShape{}, {0}),
           CreateTensor<int64>(TensorShape{}, {10}),
           CreateTensor<int64>(TensorShape{}, {1}),
           CreateTensor<int64>(TensorShape{}, {11}),
           CreateTensor<int64>(TensorShape{}, {2}),
           CreateTensor<int64>(TensorShape{}, {12})},
          /*breakpoints*/ {0, 1, 4}};
}

// Test case 2: the input datasets with different number of outputs.
TestParam TestCase2() {