Function TEST_F

tensorflow/core/kernels/data/interleave_dataset_op_test.cc:350–378

}

TEST_F(InterleaveDatasetOpTest, InvalidCycleLength) {
  int thread_num = 2, cpu_num = 2;
  const TestCase &test_case = InvalidCycleLengthTestCase();
  TF_ASSERT_OK(InitThreadPool(thread_num));
  TF_ASSERT_OK(InitFunctionLibraryRuntime(test_case.func_lib, cpu_num));

  std::unique_ptr<OpKernel> interleave_dataset_kernel;
  TF_ASSERT_OK(CreateInterleaveDatasetKernel(
      test_case.func, test_case.expected_output_dtypes,
      test_case.expected_output_shapes, &interleave_dataset_kernel));

  Tensor tensor_slice_dataset_tensor(DT_VARIANT, TensorShape({}));
  std::vector<Tensor> inputs_for_tensor_slice_dataset = test_case.input_tensors;
  TF_ASSERT_OK(CreateTensorSliceDatasetTensor(&inputs_for_tensor_slice_dataset,
                                              &tensor_slice_dataset_tensor));
  Tensor cycle_length = test_case.cycle_length;
  Tensor block_length = test_case.block_length;
  gtl::InlinedVector<TensorValue, 4> inputs(
      {TensorValue(&tensor_slice_dataset_tensor), TensorValue(&cycle_length),
       TensorValue(&block_length)});
  std::unique_ptr<OpKernelContext> interleave_dataset_context;
  TF_ASSERT_OK(CreateInterleaveDatasetContext(
      interleave_dataset_kernel.get(), &inputs, &interleave_dataset_context));
  DatasetBase *interleave_dataset;
  EXPECT_EQ(CreateDataset(interleave_dataset_kernel.get(),
                          interleave_dataset_context.get(), &interleave_dataset)
                .code(),
            tensorflow::error::INVALID_ARGUMENT);
}

TEST_F(InterleaveDatasetOpTest, InvalidBlockLength) {
  int thread_num = 2, cpu_num = 2;