From 1c88fff548acb048f2fe64fd2564e885759def12 Mon Sep 17 00:00:00 2001
From: Brian Atkinson <bca@google.com>
Date: Wed, 16 Oct 2019 14:34:03 -0700
Subject: [PATCH] Make use of ::testing::TempDir to access tmp dir.
PiperOrigin-RevId: 275115437
Change-Id: I10fcec58c9ebcc447f56306501447cab92cc29f3
---
tensorflow/lite/testing/kernel_test/BUILD | 3 ---
tensorflow/lite/testing/kernel_test/util_test.cc | 3 +--
2 files changed, 1 insertion(+), 5 deletions(-)
diff --git a/tensorflow/lite/testing/kernel_test/BUILD b/tensorflow/lite/testing/kernel_test/BUILD
index ee2403bd5c2..d1610c422b8 100644
--- a/tensorflow/lite/testing/kernel_test/BUILD
+++ b/tensorflow/lite/testing/kernel_test/BUILD
@@ -38,9 +38,6 @@ tf_cc_test(
"//tensorflow/lite:testdata/add.bin",
"//tensorflow/lite:testdata/test_input.csv",
],
- tags = [
- "no_oss",
- ],
deps = [
":util",
"//tensorflow/lite/testing:tflite_driver",
diff --git a/tensorflow/lite/testing/kernel_test/util_test.cc b/tensorflow/lite/testing/kernel_test/util_test.cc
index 0599ded72d8..d9d8391bba1 100644
--- a/tensorflow/lite/testing/kernel_test/util_test.cc
+++ b/tensorflow/lite/testing/kernel_test/util_test.cc
@@ -18,7 +18,6 @@ limitations under the License.
#include <memory>
#include <gmock/gmock.h>
-#include "testing/base/public/googletest.h"
#include <gtest/gtest.h>
#include "tensorflow/lite/testing/tflite_driver.h"
@@ -32,7 +31,7 @@ TEST(UtilTest, SimpleE2ETest) {
options.tflite_model = "tensorflow/lite/testdata/add.bin";
options.read_input_from_file =
"tensorflow/lite/testdata/test_input.csv";
- options.dump_output_to_file = FLAGS_test_tmpdir + "/test_out.csv";
+ options.dump_output_to_file = ::testing::TempDir() + "/test_out.csv";
options.kernel_type = "REFERENCE";
std::unique_ptr<TestRunner> runner(new TfLiteDriver(
TfLiteDriver::DelegateType::kNone, /*reference_kernel=*/true));