From 989f292f20b21ae0bf379d686ef11af9645fd2a9 Mon Sep 17 00:00:00 2001
From: spiglerg <spiglerg@gmail.com>
Date: Wed, 10 Apr 2019 11:40:10 +0200
Subject: [PATCH] new rendering tests
---
tests/test_rendertools.py | 12 ++++++------
1 file changed, 6 insertions(+), 6 deletions(-)
diff --git a/tests/test_rendertools.py b/tests/test_rendertools.py
index 39430d67..b73b892a 100644
--- a/tests/test_rendertools.py
+++ b/tests/test_rendertools.py
@@ -39,12 +39,12 @@ def checkFrozenImage(sFileImage):
bytesFrozenImage = None
for sDir in [ "/images/", "/images/test/" ]:
sfPath = sDirRoot + sDir + sFileImage
- with open(sfPath, "rb") as fIn:
- bytesImage = fIn.read()
- if bytesFrozenImage == None:
- bytesFrozenImage = bytesImage
- else:
- assert(bytesFrozenImage == bytesImage)
+ bytesImage = plt.imread(sfPath)
+ if bytesFrozenImage is None:
+ bytesFrozenImage = bytesImage
+ else:
+ assert(bytesFrozenImage.shape == bytesImage.shape)
+ assert((np.sum(np.square(bytesFrozenImage-bytesImage)) / bytesFrozenImage.size) < 1e-3)
def test_render_env():
--
GitLab