Function TestCase2

tensorflow/core/kernels/data/window_dataset_op_test.cc:93–111

Test case 2: size=2, shift=2, stride=2, drop_remainder=true.

// Test case 2: size=2, shift=2, stride=2, drop_remainder=true.
TestCase TestCase2() {
  return {/*range_data_param*/ {0, 7, 1},
          /*size*/ CreateTensor<int64>(TensorShape({}), {2}),
          /*shift*/ CreateTensor<int64>(TensorShape({}), {2}),
          /*stride*/ CreateTensor<int64>(TensorShape({}), {2}),
          /*drop_remainder*/
          CreateTensor<bool>(TensorShape({}), {true}),
          /*expected_outputs*/
          {{CreateTensor<int64>(TensorShape({}), {0}),
            CreateTensor<int64>(TensorShape({}), {2})},
           {CreateTensor<int64>(TensorShape({}), {2}),
            CreateTensor<int64>(TensorShape({}), {4})},
           {CreateTensor<int64>(TensorShape({}), {4}),
            CreateTensor<int64>(TensorShape({}), {6})}},
          /*expected_output_dtypes*/ {DT_VARIANT},
          /*expected_output_shapes*/ {PartialTensorShape({})},
          /*expected_cardinality*/ 3,
          /*breakpoints*/ {0, 1, 9}};
}

// Test case 3: size=8, shift=3, stride=1, drop_remainder=false.
TestCase TestCase3() {