Function TestCase5

tensorflow/core/kernels/data/interleave_dataset_op_test.cc:186–206

test case 5: cycle_length = 2, block_length = 2.

// test case 5: cycle_length = 2, block_length = 2.
TestCase TestCase5() {
  return {/*input_tensors*/
          {CreateTensor<tstring>(TensorShape{3, 3, 1}, {"a", "b", "c", "d", "e",
                                                        "f", "g", "h", "i"})},
          /*func*/
          MakeTensorSliceDatasetFunc(
              DataTypeVector({DT_STRING}),
              std::vector<PartialTensorShape>({PartialTensorShape({1})})),
          /*func_lib*/ {test::function::MakeTensorSliceDataset()},
          /*cycle_length*/
          CreateTensor<int64>(TensorShape({}), {2}),
          /*block_length*/
          CreateTensor<int64>(TensorShape({}), {2}),
          /*expected_outputs*/
          ConvertToTensorVec<tstring>(
              {"a", "b", "d", "e", "c", "f", "g", "h", "i"}),
          /*expected_output_dtypes*/ {DT_STRING},
          /*expected_output_shapes*/ {PartialTensorShape({1})},
          /*expected_cardinality*/ tensorflow::data::kUnknownCardinality,
          /*breakpoints*/ {0, 4, 11}};
}

// test case 6: cycle_length = 2, block_length = 3.
TestCase TestCase6() {