Function TestCase2

tensorflow/core/kernels/data/interleave_dataset_op_test.cc:120–139

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

// test case 2: cycle_length = 2, block_length = 1.
TestCase TestCase2() {
  return {
      /*input_tensors*/
      {CreateTensor<int64>(TensorShape{3, 3, 1}, {0, 1, 2, 3, 4, 5, 6, 7, 8})},
      /*func*/
      MakeTensorSliceDatasetFunc(
          DataTypeVector({DT_INT64}),
          std::vector<PartialTensorShape>({PartialTensorShape({1})})),
      /*func_lib*/ {test::function::MakeTensorSliceDataset()},
      /*cycle_length*/
      CreateTensor<int64>(TensorShape({}), {2}),
      /*block_length*/
      CreateTensor<int64>(TensorShape({}), {1}),
      /*expected_outputs*/
      ConvertToTensorVec<int64>({0, 3, 1, 4, 2, 5, 6, 7, 8}),
      /*expected_output_dtypes*/ {DT_INT64},
      /*expected_output_shapes*/ {PartialTensorShape({1})},
      /*expected_cardinality*/ tensorflow::data::kUnknownCardinality,
      /*breakpoints*/ {0, 4, 11}};
}

// test case 3: cycle_length = 3, block_length = 1.
TestCase TestCase3() {