Function TEST_F

tensorflow/core/kernels/data/map_defun_op_test.cc:222–248

                             {TestCase1(), TestCase2(), TestCase3()})));

TEST_F(MapDefunOpTest, InvalidArguments) {
  int thread_num = 2, cpu_num = 2;
  TF_ASSERT_OK(InitThreadPool(thread_num));
  std::vector<TestCase> test_cases = {InvalidOutputTypes(),
                                      InvalidOutputShapes(), InvalidInputs()};
  for (auto& test_case : test_cases) {
    TF_ASSERT_OK(InitFunctionLibraryRuntime(test_case.func_lib, cpu_num));

    std::unique_ptr<OpKernel> map_defun_kernel;
    TF_ASSERT_OK(CreateMapDefunOpKernel(
        test_case.t_arguments, test_case.t_captured, test_case.output_dtypes,
        test_case.output_shapes, test_case.func,
        test_case.max_intra_op_parallelism, &map_defun_kernel));
    gtl::InlinedVector<TensorValue, 4> inputs;
    for (auto& arg : test_case.arguments) {
      inputs.emplace_back(&arg);
    }
    for (auto& captured_input : test_case.captured_inputs) {
      inputs.emplace_back(&captured_input);
    }
    std::unique_ptr<OpKernelContext> context;
    TF_ASSERT_OK(
        CreateMapDefunContext(map_defun_kernel.get(), &inputs, &context));
    EXPECT_EQ(RunOpKernel(map_defun_kernel.get(), context.get()).code(),
              tensorflow::error::INVALID_ARGUMENT);
  }
}

}  // namespace
}  // namespace data