diff --git a/GPy/testing/plotting_tests.py b/GPy/testing/plotting_tests.py index 28eb5c8d..e60928a7 100644 --- a/GPy/testing/plotting_tests.py +++ b/GPy/testing/plotting_tests.py @@ -101,7 +101,9 @@ def test_kernel(): matplotlib.rcParams[u'text.usetex'] = False k = GPy.kern.RBF(5, ARD=True) + GPy.kern.Linear(3, active_dims=[0,2,4], ARD=True) + GPy.kern.Bias(2) k.randomize() - k.plot_ARD(legend=True) + k2 = GPy.kern.RBF(5, ARD=True) + GPy.kern.Linear(3, active_dims=[0,2,4], ARD=True) + GPy.kern.Bias(2) + GPy.kern.White(4) + k2[:-1] = k[:] + k2.plot_ARD([k2.white.name], legend=True) for do_test in _image_comparison( baseline_images=['kern_{}'.format(sub) for sub in ["ARD",]], extensions=extensions):