add test

Author: ZJY0516

tests/utils_/test_utils.py

@@ -30,7 +30,8 @@
                         get_tcp_uri, is_lossless_cast, join_host_port,
                         make_zmq_path, make_zmq_socket, memory_profiling,
                         merge_async_iterators, sha256, split_host_port,
-                        split_zmq_path, supports_kw, swap_dict_values)
+                        split_zmq_path, supports_kw, swap_dict_values,
+                        unique_filepath)
 
 from ..utils import create_new_process_for_each_test, error_on_warning
 
@@ -1032,3 +1033,17 @@ def test_load_config_file(tmp_path):
     # Assert that the processed arguments match the expected output
     assert processed_args == expected_args
     os.remove(str(config_file_path))
+
+
+def test_unique_filepath():
+    # make temp directory
+    temp_dir = tempfile.mkdtemp()
+    path_fn = lambda i: Path(temp_dir) / f"file_{i}.txt"
+    paths = set()
+    for i in range(10):
+        path = unique_filepath(path_fn)
+        # write to file
+        path.write_text("test")
+        paths.add(path)
+    assert len(paths) == 10
+    assert len(list(Path(temp_dir).glob("*.txt"))) == 10