diff --git a/tests/test_merge.cpp b/tests/test_merge.cpp index cd13f907d4..9bb43824eb 100644 --- a/tests/test_merge.cpp +++ b/tests/test_merge.cpp @@ -60,7 +60,7 @@ struct CommonData { }; CommonData cd; - +std::string temp_filename_template = "/tmp/faiss_tmp_XXXXXX"; /// perform a search on shards, then merge and search again and /// compare results. int compare_merged( @@ -71,7 +71,7 @@ int compare_merged( std::vector refD(k * nq); index_shards->search(nq, cd.queries.data(), k, refD.data(), refI.data()); - Tempfilename filename(&temp_file_mutex, "/tmp/faiss_tmp_XXXXXX"); + Tempfilename filename(&temp_file_mutex, temp_filename_template); std::vector newI(k * nq); std::vector newD(k * nq); @@ -191,7 +191,7 @@ TEST(MERGE, merge_flat_vt) { TEST(MERGE, merge_flat_ondisk) { faiss::IndexShards index_shards(d, false, false); index_shards.own_indices = true; - Tempfilename filename(&temp_file_mutex, "/tmp/faiss_tmp_XXXXXX"); + Tempfilename filename(&temp_file_mutex, temp_filename_template); for (int i = 0; i < nindex; i++) { auto ivf = new faiss::IndexIVFFlat(&cd.quantizer, d, nlist); diff --git a/tests/test_util.h b/tests/test_util.h index ff76ed3e9c..75ff9afcbf 100644 --- a/tests/test_util.h +++ b/tests/test_util.h @@ -10,17 +10,16 @@ #include #include -#include struct Tempfilename { pthread_mutex_t* mutex; std::string filename; - Tempfilename(pthread_mutex_t* mutex, std::string filename) { + Tempfilename(pthread_mutex_t* mutex, std::string filename_template) { this->mutex = mutex; - this->filename = filename; + this->filename = filename_template; pthread_mutex_lock(mutex); - int fd = mkstemp(&filename[0]); + int fd = mkstemp(&this->filename[0]); close(fd); pthread_mutex_unlock(mutex); }