Automated Code Change

PiperOrigin-RevId: 611562592
tensorflow · Feb 29, 2024 · 3c280ea · 3c280ea
1 parent e8bda99
commit 3c280ea
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 12 deletions.
diff --git a/tensorflow_lite_support/scann_ondevice/cc/test/BUILD b/tensorflow_lite_support/scann_ondevice/cc/test/BUILD
@@ -30,7 +30,7 @@ cc_test(
         "//tensorflow_lite_support/scann_ondevice/cc:index",
         "//tensorflow_lite_support/scann_ondevice/cc/core:serialized_searcher_cc_proto",
         "//tensorflow_lite_support/scann_ondevice/proto:index_config_cc_proto",
-        "@com_google_absl//absl/flags:flag",
+        "//testing/base/public:gunit",
         "@com_google_absl//absl/status",
         "@com_google_absl//absl/strings",
         "@org_tensorflow//tensorflow/lite:test_util",
@@ -54,7 +54,7 @@ cc_test(
         "//tensorflow_lite_support/cc/test:test_utils",
         "//tensorflow_lite_support/scann_ondevice/cc:index_builder",
         "//tensorflow_lite_support/scann_ondevice/proto:index_config_cc_proto",
-        "@com_google_absl//absl/flags:flag",
+        "//testing/base/public:gunit",
         "@com_google_absl//absl/memory",
         "@com_google_absl//absl/status",
         "@com_google_absl//absl/strings",

diff --git a/tensorflow_lite_support/scann_ondevice/cc/test/index_builder_test.cc b/tensorflow_lite_support/scann_ondevice/cc/test/index_builder_test.cc
@@ -18,7 +18,7 @@ limitations under the License.
 #include <cstdint>
 #include <string>
 
-#include "absl/flags/flag.h"  // from @com_google_absl
+#include <gtest/gtest.h>
 #include "absl/memory/memory.h"  // from @com_google_absl
 #include "absl/status/status.h"  // from @com_google_absl
 #include "absl/strings/str_format.h"  // from @com_google_absl
@@ -132,7 +132,7 @@ class PopulateIndexFileTest : public TestWithParam<bool /*compression*/> {};
 
 TEST_P(PopulateIndexFileTest, WritesHashedDatabaseWithPartitioner) {
   const std::string db_path =
-      tflite::task::JoinPath(getenv("TEST_TMPDIR"), "hashed");
+      tflite::task::JoinPath(::testing::TempDir(), "hashed");
   const bool compression = GetParam();
 
   {
@@ -248,7 +248,7 @@ TEST_P(PopulateIndexFileTest, WritesHashedDatabaseWithPartitioner) {
 
 TEST_P(PopulateIndexFileTest, WritesHashedDatabaseWithoutPartitioner) {
   const std::string db_path =
-      tflite::task::JoinPath(getenv("TEST_TMPDIR"), "float");
+      tflite::task::JoinPath(::testing::TempDir(), "float");
   const bool compression = GetParam();
 
   {
@@ -337,7 +337,7 @@ TEST_P(PopulateIndexFileTest, WritesHashedDatabaseWithoutPartitioner) {
 
 TEST_P(PopulateIndexFileTest, WritesFloatDatabaseWithPartitioner) {
   const std::string db_path =
-      tflite::task::JoinPath(getenv("TEST_TMPDIR"), "float");
+      tflite::task::JoinPath(::testing::TempDir(), "float");
   const bool compression = GetParam();
 
   {
@@ -456,7 +456,7 @@ TEST_P(PopulateIndexFileTest, WritesFloatDatabaseWithPartitioner) {
 
 TEST_P(PopulateIndexFileTest, WritesFloatDatabaseWithoutPartitioner) {
   const std::string db_path =
-      tflite::task::JoinPath(getenv("TEST_TMPDIR"), "float");
+      tflite::task::JoinPath(::testing::TempDir(), "float");
   const bool compression = GetParam();
 
   {

diff --git a/tensorflow_lite_support/scann_ondevice/cc/test/index_test.cc b/tensorflow_lite_support/scann_ondevice/cc/test/index_test.cc
@@ -19,7 +19,7 @@ limitations under the License.
 #include <memory>
 
 #include "tensorflow_lite_support/scann_ondevice/cc/core/serialized_searcher.pb.h"
-#include "absl/flags/flag.h"  // from @com_google_absl
+#include <gtest/gtest.h>
 #include "absl/status/status.h"  // from @com_google_absl
 #include "absl/strings/string_view.h"  // from @com_google_absl
 #include "tensorflow/lite/test_util.h"
@@ -47,8 +47,7 @@ constexpr char kDummyIndexPath[] =
 TEST(CreateFromOptionsTest, Succeeds) {
   // Load file in memory using ExternalFile.
   ExternalFile file;
-  file.set_file_name(
-      JoinPath("./" /*test src dir*/, kDummyIndexPath));
+  file.set_file_name(JoinPath(::testing::SrcDir(), kDummyIndexPath));
   SUPPORT_ASSERT_OK_AND_ASSIGN(std::unique_ptr<ExternalFileHandler> handler,
                        ExternalFileHandler::CreateFromExternalFile(&file));
   absl::string_view file_contents = handler->GetFileContent();
@@ -62,8 +61,7 @@ class IndexTest : public tflite::testing::Test {
   IndexTest() {
     // Load file in memory using ExternalFile.
     ExternalFile file;
-    file.set_file_name(
-        JoinPath("./" /*test src dir*/, kDummyIndexPath));
+    file.set_file_name(JoinPath(::testing::SrcDir(), kDummyIndexPath));
     handler_ = ExternalFileHandler::CreateFromExternalFile(&file).value();
     absl::string_view file_contents = handler_->GetFileContent();
     // Build index.