From 4f6d21d315cb1d6889eb0cff04e5b99cd6937413 Mon Sep 17 00:00:00 2001 From: Gilles DAVID Date: Mon, 13 Feb 2023 14:35:23 +0100 Subject: [PATCH] Use tempfile.mkdtemp to create temporary directory The function is cross-platform. Moreover, the variable LOGNAME does not exist in a Docker container --- src/pmml/pmml_swig/PMMLBasicsTest.py | 12 ++---------- 1 file changed, 2 insertions(+), 10 deletions(-) diff --git a/src/pmml/pmml_swig/PMMLBasicsTest.py b/src/pmml/pmml_swig/PMMLBasicsTest.py index 38a0945f9..3cbe2a265 100644 --- a/src/pmml/pmml_swig/PMMLBasicsTest.py +++ b/src/pmml/pmml_swig/PMMLBasicsTest.py @@ -27,21 +27,13 @@ import unittest import os import shutil import platform +import tempfile class PMMLBasicsTest(unittest.TestCase): def setUp(self): self.resourcesDir = ".." + os.sep + "Test" + os.sep + "samples" + os.sep ; - if platform.system() == "Windows" : - self.tmpDir = os.environ['TMP'] # %TMP% does exist on WINDOWS - else: - self.tmpDir = os.sep + "tmp" + os.sep + os.environ['LOGNAME'] + os.sep ; - self.tmpDir += "PmmlUnitTest"; - self.tmpDir += os.sep ; - if ( not os.path.exists(self.tmpDir) ): - os.makedirs(self.tmpDir); - pass - pass + self.tmpDir = tempfile.mkdtemp(suffix="PmmlUnitTest") def tearDown(self): if ( os.path.exists(self.tmpDir) ): -- 2.39.2