Use tempfile for temporary directory in tests. (#708)

lanpa · Jul 16, 2023 · c015fea · c015fea
1 parent a7013c3
commit c015fea
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/test_summary_writer.py b/tests/test_summary_writer.py
@@ -1,4 +1,5 @@
 from tensorboardX import SummaryWriter
+import tempfile
 import unittest
 
 
@@ -9,9 +10,10 @@ def test_summary_writer_ctx(self):
             writer.add_scalar('test', 1)
         assert writer.file_writer is None
 
-    def test_summary_writer_backcomapt(self):
-        with SummaryWriter(log_dir='/tmp/tbxtest') as writer:
-            writer.add_scalar('test', 1)
+    def test_summary_writer_backcompat(self):
+        with tempfile.TemporaryDirectory() as tmp_dir:
+            with SummaryWriter(log_dir=tmp_dir) as writer:
+                writer.add_scalar('test', 1)
 
     def test_summary_writer_close(self):
         # Opening and closing SummaryWriter a lot should not run into