diff --git a/tensorflow/core/kernels/data/text_line_dataset_op_test.cc b/tensorflow/core/kernels/data/text_line_dataset_op_test.cc index f4c9589856d..e3f6e739ea8 100644 --- a/tensorflow/core/kernels/data/text_line_dataset_op_test.cc +++ b/tensorflow/core/kernels/data/text_line_dataset_op_test.cc @@ -19,6 +19,12 @@ namespace { constexpr char kNodeName[] = "text_line_dataset"; +tstring LocalTempFilename() { + std::string path; + CHECK(Env::Default()->LocalTempFilename(&path)); + return tstring(path); +} + class TextLineDatasetParams : public DatasetParams { public: TextLineDatasetParams(std::vector filenames, @@ -84,9 +90,7 @@ Status CreateTestFiles(const std::vector& filenames, // Test case 1: multiple text files with ZLIB compression. TextLineDatasetParams TextLineDatasetParams1() { - std::vector filenames = { - absl::StrCat(testing::TmpDir(), "/text_line_ZLIB_1"), - absl::StrCat(testing::TmpDir(), "/text_line_ZLIB_2")}; + std::vector filenames = {LocalTempFilename(), LocalTempFilename()}; std::vector contents = { absl::StrCat("hello world\n", "11223334455\n"), absl::StrCat("abcd, EFgH\n", " \n", "$%^&*()\n")}; @@ -103,9 +107,7 @@ TextLineDatasetParams TextLineDatasetParams1() { // Test case 2: multiple text files with GZIP compression. TextLineDatasetParams TextLineDatasetParams2() { - std::vector filenames = { - absl::StrCat(testing::TmpDir(), "/text_line_GZIP_1"), - absl::StrCat(testing::TmpDir(), "/text_line_GZIP_2")}; + std::vector filenames = {LocalTempFilename(), LocalTempFilename()}; std::vector contents = { absl::StrCat("hello world\n", "11223334455\n"), absl::StrCat("abcd, EFgH\n", " \n", "$%^&*()\n")}; @@ -122,9 +124,7 @@ TextLineDatasetParams TextLineDatasetParams2() { // Test case 3: multiple text files without compression. TextLineDatasetParams TextLineDatasetParams3() { - std::vector filenames = { - absl::StrCat(testing::TmpDir(), "/text_line_UNCOMPRESSED_1"), - absl::StrCat(testing::TmpDir(), "/text_line_UNCOMPRESSED_2")}; + std::vector filenames = {LocalTempFilename(), LocalTempFilename()}; std::vector contents = { absl::StrCat("hello world\n", "11223334455\n"), absl::StrCat("abcd, EFgH\n", " \n", "$%^&*()\n")};