From 0be0e27fef4cd6425aab712ae9cc061bc5f54a32 Mon Sep 17 00:00:00 2001 From: hagrid67 <jdhwatson@gmail.com> Date: Tue, 9 Apr 2019 22:22:40 +0100 Subject: [PATCH] tweaked root folder for images --- tests/test_rendertools.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/test_rendertools.py b/tests/test_rendertools.py index bfe8f620..39430d67 100644 --- a/tests/test_rendertools.py +++ b/tests/test_rendertools.py @@ -28,7 +28,8 @@ import flatland.utils.rendertools as rt def checkFrozenImage(sFileImage): - sTmpFileImage = "flatland/images/test/" + sFileImage + sDirRoot = "." + sTmpFileImage = sDirRoot + "/images/test/" + sFileImage if os.path.exists(sTmpFileImage): os.remove(sTmpFileImage) @@ -36,8 +37,8 @@ def checkFrozenImage(sFileImage): plt.savefig(sTmpFileImage) bytesFrozenImage = None - for sDir in [ "flatland/images/", "flatland/images/test/" ]: - sfPath = sDir + sFileImage + for sDir in [ "/images/", "/images/test/" ]: + sfPath = sDirRoot + sDir + sFileImage with open(sfPath, "rb") as fIn: bytesImage = fIn.read() if bytesFrozenImage == None: -- GitLab