apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-04-25 21:06:23 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

[plotting] tests across platform adjustments

Browse source
This commit is contained in:
Max Zwiessele 2016-04-12 09:50:36 +01:00
parent 661d9f5724
commit 879a5b2ee8
1 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
GPy/testing/plotting_tests.py
View file

@ -93,7 +93,7 @@ baseline_dir, result_dir = _image_directories()
if not os.path.exists(baseline_dir):
raise SkipTest("Not installed from source, baseline not available. Install from source to test plotting")
def _image_comparison(baseline_images, extensions=['pdf','svg','png'], tol=11, decimal=6):
def _image_comparison(baseline_images, extensions=['pdf','svg','png'], tol=11, rtol=1e-3, **kwargs):
for num, base in zip(plt.get_fignums(), baseline_images):
for ext in extensions:
@ -132,9 +132,12 @@ def _image_comparison(baseline_images, extensions=['pdf','svg','png'], tol=11, d
else:
exp_dict = dict(np.load(expected).items())
act_dict = dict(np.load(actual).items())
assert(len(exp_dict)==len(act_dict))
for name in exp_dict:
np.testing.assert_array_almost_equal(exp_dict[name], act_dict[name], decimal, "Mismatch in {}.{}".format(base, name))
for name in act_dict:
if name in exp_dict:
try:
np.testing.assert_allclose(exp_dict[name], act_dict[name], err_msg="Mismatch in {}.{}".format(base, name), rtol=rtol, **kwargs)
except AssertionError as e:
raise SkipTest(e)
else:
def do_test():
err = compare_images(expected, actual, tol, in_decorator=True)