Function TestCase1

tensorflow/core/kernels/data/cache_dataset_ops_test.cc:99–111

Test case 1: cache data in file.

// Test case 1: cache data in file.
TestCase TestCase1() {
  return {/*input_tensors*/ {CreateTensor<int64>(TensorShape{3, 3, 1},
                                                 {0, 1, 2, 3, 4, 5, 6, 7, 8})},
          /*file_name*/ absl::StrCat(testing::TmpDir(), "/cache_data"),
          /*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 2: cache empty data in file.
TestCase TestCase2() {