apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-07-14 16:32:15 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

[plotting] gradient plot added

Browse source
This commit is contained in:
mzwiessele 2015-10-05 19:08:38 +01:00
parent 0880948f3b
commit d8901f75bb
1 changed files with 17 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

34
GPy/testing/plotting_tests.py
View file

@ -75,23 +75,23 @@ def _image_comparison(baseline_images, extensions=['pdf','svg','ong'], tol=1e-3)
fig.set_frameon(False) fig.set_frameon(False)
fig.canvas.draw() fig.canvas.draw()
fig.savefig(os.path.join(result_dir, "{}.{}".format(base, ext))) fig.savefig(os.path.join(result_dir, "{}.{}".format(base, ext)))
for num, base in zip(plt.get_fignums(), baseline_images): for num, base in zip(plt.get_fignums(), baseline_images):
for ext in extensions: for ext in extensions:
#plt.close(num) #plt.close(num)
actual = os.path.join(result_dir, "{}.{}".format(base, ext)) actual = os.path.join(result_dir, "{}.{}".format(base, ext))
expected = os.path.join(baseline_dir, "{}.{}".format(base, ext)) expected = os.path.join(baseline_dir, "{}.{}".format(base, ext))
def do_test(): def do_test():
err = compare_images(expected, actual, tol) err = compare_images(expected, actual, tol)
try: try:
if not os.path.exists(expected): if not os.path.exists(expected):
raise ImageComparisonFailure( raise ImageComparisonFailure(
'image does not exist: %s' % expected) 'image does not exist: %s' % expected)
if err: if err:
raise ImageComparisonFailure( raise ImageComparisonFailure(
'images not close: {err[actual]!s} vs. {err[expected]!s} (RMS {err[rms]:.3f})'.format(err=err)) 'images not close: {err[actual]!s} vs. {err[expected]!s} (RMS {err[rms]:.3f})'.format(err=err))
except ImageComparisonFailure: except ImageComparisonFailure:
pass pass
yield do_test yield do_test
plt.close('all') plt.close('all')
def test_plot(self=None): def test_plot(self=None):