commit aa22d6aa35cf589633323c0aad2c235f69549a3e
author Renjie Liu <renjieliu@google.com> Wed Sep 02 19:11:31 2020 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Wed Sep 02 19:17:16 2020 -0700
tree a9c901e48bfce6ce50868a12ecf3008e530054b1
parent c658e2fc825c07c06878633fff6c1c8feace2495

Remove main function in the custom op tests.

PiperOrigin-RevId: 329834644
Change-Id: I28e07d24071371dacf7c71d7728de7a54047baf2

tensorflow/lite/kernels/BUILD

diff --git a/tensorflow/lite/kernels/BUILD b/tensorflow/lite/kernels/BUILD
index 45a1c97..7d32acf 100644
--- a/tensorflow/lite/kernels/BUILD
+++ b/tensorflow/lite/kernels/BUILD
@@ -2140,6 +2140,7 @@
     srcs = ["rfft2d_test.cc"],
     deps = [
         ":custom_ops",
+        ":test_main",
         ":test_util",
         "//tensorflow/lite:framework",
         "//tensorflow/lite/c:common",

tensorflow/lite/kernels/complex_support_test.cc

diff --git a/tensorflow/lite/kernels/complex_support_test.cc b/tensorflow/lite/kernels/complex_support_test.cc
index cb60345..20f8867 100644
--- a/tensorflow/lite/kernels/complex_support_test.cc
+++ b/tensorflow/lite/kernels/complex_support_test.cc
@@ -159,9 +159,3 @@
 }  // namespace custom
 }  // namespace ops
 }  // namespace tflite
-
-int main(int argc, char** argv) {
-  ::tflite::LogToStderr();
-  ::testing::InitGoogleTest(&argc, argv);
-  return RUN_ALL_TESTS();
-}

tensorflow/lite/kernels/rfft2d_test.cc

diff --git a/tensorflow/lite/kernels/rfft2d_test.cc b/tensorflow/lite/kernels/rfft2d_test.cc
index e9b23ba..e7d806a 100644
--- a/tensorflow/lite/kernels/rfft2d_test.cc
+++ b/tensorflow/lite/kernels/rfft2d_test.cc
@@ -150,9 +150,3 @@
 }  // namespace custom
 }  // namespace ops
 }  // namespace tflite
-
-int main(int argc, char** argv) {
-  ::tflite::LogToStderr();
-  ::testing::InitGoogleTest(&argc, argv);
-  return RUN_ALL_TESTS();
-}