diff --git a/jass/test/test_plots.py b/jass/test/test_plots.py index 390dbe614c8dbaabe1f5fbb9b6eda95b5902bb35..f3891c3f68efa6b3a7ad5894823872fd5e577818 100644 --- a/jass/test/test_plots.py +++ b/jass/test/test_plots.py @@ -3,8 +3,10 @@ from __future__ import absolute_import import os, shutil, tempfile from pathlib import Path - +import matplotlib as plt from jass.models import plots +from PIL import Image +import numpy as np from . import JassTestCase @@ -13,6 +15,7 @@ class TestPlots(JassTestCase): test_folder = "data_real" expected_res_folder = "expected_graphs" + #expected_res_folder="baseline_images/test_plot" def setUp(self): # Create a temporary directory @@ -26,15 +29,23 @@ class TestPlots(JassTestCase): pass def test_create_global_plot(self): + #import shutil + #print(plt.rcParams) plots.create_global_plot(self.worktable_hdf_path, self.test_dir / "global_plot.png") #plots.create_global_plot(self.worktable_hdf_path, "/Users/vlegrand/tmp/global_plot.png") - fnew=open(self.test_dir / "global_plot.png",'rb') - fref=open(self.ref_res_dir / "expected_global_plot.png",'rb') - content_new=fnew.read() - content_ref=fref.read() - assert(content_new==content_ref) - fnew.close() - fref.close() + img_new=Image.open(self.test_dir /"global_plot.png") + #fnew=open(self.test_dir / "global_plot.png",'rb') + #fref=open(self.ref_res_dir / "expected_global_plot.png",'rb') + img_ref=Image.open(self.ref_res_dir / "expected_global_plot.png") + sum_sq_diff = np.sum((np.asarray(img_new).astype('float') - np.asarray(img_ref).astype('float')) ** 2) + print("sum_sq_diff=",sum_sq_diff) + assert(sum_sq_diff==0.0) + #content_new=fnew.read() + #content_ref=fref.read() + shutil.copyfile(self.test_dir / "global_plot.png","/pasteur/gaia/homes/vlegrand/global_plot.png") + #assert(content_new==content_ref) + #fnew.close() + #fref.close() def test_create_qq_plot(self):