Python: Use a temporary directory for generated files in tests (#481)

google · Dec 12, 2016 · 6f22722 · 6f22722
1 parent 4651f7c
commit 6f22722
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/python/tests/_test_utils.py b/python/tests/_test_utils.py
@@ -5,6 +5,7 @@
 import os
 import sys
 import sysconfig
+import tempfile
 import unittest
 
 
@@ -44,13 +45,15 @@
 TESTDATA_PATHS_FOR_DECOMPRESSION = glob.glob(
     os.path.join(TESTDATA_DIR, '*.compressed'))
 
+TEMP_DIR = tempfile.mkdtemp()
+
 
 def get_temp_compressed_name(filename):
-    return filename + '.bro'
+    return os.path.join(TEMP_DIR, os.path.basename(filename + '.bro'))
 
 
 def get_temp_uncompressed_name(filename):
-    return filename + '.unbro'
+    return os.path.join(TEMP_DIR, os.path.basename(filename + '.unbro'))
 
 
 def bind_method_args(method, *args, **kwargs):