Function TestCase3

tensorflow/core/kernels/data/cache_dataset_ops_test.cc:125–137

Test case 3: cache data in memory.

// Test case 3: cache data in memory.
TestCase TestCase3() {
  return {/*input_tensors*/ {CreateTensor<int64>(TensorShape{3, 3, 1},
                                                 {0, 1, 2, 3, 4, 5, 6, 7, 8})},
          /*file_name*/ "",
          /*expected_outputs*/
          {CreateTensor<int64>(TensorShape{3, 1}, {0, 1, 2}),
           CreateTensor<int64>(TensorShape{3, 1}, {3, 4, 5}),
           CreateTensor<int64>(TensorShape{3, 1}, {6, 7, 8})},
          /*expected_output_dtypes*/ {DT_INT64},
          /*expected_output_shapes*/ {PartialTensorShape({3, 1})},
          /*expected_cardinality*/ 3,
          /*breakpoints*/ {0, 2, 4, 11}};
}

// Test case 4: cache empty data in memory.
TestCase TestCase4() {