From db5816707ab0fc4ae6d3959a313d67dd871a8d9d Mon Sep 17 00:00:00 2001
From: Brian Atkinson <bca@google.com>
Date: Wed, 16 Oct 2019 14:44:56 -0700
Subject: [PATCH] Make use of ::testing::TempDir() to access a tmp dir.
PiperOrigin-RevId: 275117892
Change-Id: I99a2f9489df322a4845eb0c9aab3df2f5581684a
---
tensorflow/lite/testing/kernel_test/BUILD | 3 ---
tensorflow/lite/testing/kernel_test/input_generator_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 d1610c422b8..1a2e22c921e 100644
--- a/tensorflow/lite/testing/kernel_test/BUILD
+++ b/tensorflow/lite/testing/kernel_test/BUILD
@@ -84,9 +84,6 @@ cc_test(
"//tensorflow/lite:testdata/multi_add.bin",
"//tensorflow/lite:testdata/test_input.csv",
],
- tags = [
- "no_oss",
- ],
deps = [
":input_generator",
"@com_google_googletest//:gtest_main",
diff --git a/tensorflow/lite/testing/kernel_test/input_generator_test.cc b/tensorflow/lite/testing/kernel_test/input_generator_test.cc
index 2223a9196ff..1719aa7bc96 100644
--- a/tensorflow/lite/testing/kernel_test/input_generator_test.cc
+++ b/tensorflow/lite/testing/kernel_test/input_generator_test.cc
@@ -18,7 +18,6 @@ limitations under the License.
#include <map>
#include <gmock/gmock.h>
-#include "testing/base/public/googletest.h"
#include <gtest/gtest.h>
namespace tflite {
@@ -47,7 +46,7 @@ TEST(InputGeneratorTest, ReadWriteSimpleFile) {
inputs.push_back(content);
ASSERT_EQ(input_generator.GetInputs(), inputs);
- auto output_filename = FLAGS_test_tmpdir + "/out.csv";
+ auto output_filename = ::testing::TempDir() + "/out.csv";
ASSERT_EQ(input_generator.WriteInputsToFile(output_filename), kTfLiteOk);
std::ifstream in(output_filename);