fix: plotting_test

GPy/likelihoods/likelihood.py

@@ -296,7 +296,7 @@ class Likelihood(Parameterized):
             elif quad_mode == 'gh':
                 f = partial(self.integrate_gh)
                 quads = zip(*map(f, Y.flatten(), mu.flatten(), np.sqrt(sigma2.flatten())))
-                quads = np.hstack(quads)
+                quads = np.hstack(list(quads))
                 quads = quads.T
             else:
                 raise Exception("no other quadrature mode available")

GPy/plotting/matplot_dep/base_plots.py

@@ -3,6 +3,8 @@
 from matplotlib import pyplot as plt
 import numpy as np
 
+from .util import align_subplot_array, align_subplots
+
 def ax_default(fignum, ax):
     if ax is None:
         fig = plt.figure(fignum)
@@ -167,65 +169,6 @@ def fewerXticks(ax=None,divideby=2):
     ax = ax or plt.gca()
     ax.set_xticks(ax.get_xticks()[::divideby])
 
-def align_subplots(N,M,xlim=None, ylim=None):
-    """make all of the subplots have the same limits, turn off unnecessary ticks"""
-    #find sensible xlim,ylim
-    if xlim is None:
-        xlim = [np.inf,-np.inf]
-        for i in range(N*M):
-            plt.subplot(N,M,i+1)
-            xlim[0] = min(xlim[0],plt.xlim()[0])
-            xlim[1] = max(xlim[1],plt.xlim()[1])
-    if ylim is None:
-        ylim = [np.inf,-np.inf]
-        for i in range(N*M):
-            plt.subplot(N,M,i+1)
-            ylim[0] = min(ylim[0],plt.ylim()[0])
-            ylim[1] = max(ylim[1],plt.ylim()[1])
-
-    for i in range(N*M):
-        plt.subplot(N,M,i+1)
-        plt.xlim(xlim)
-        plt.ylim(ylim)
-        if (i)%M:
-            plt.yticks([])
-        else:
-            removeRightTicks()
-        if i<(M*(N-1)):
-            plt.xticks([])
-        else:
-            removeUpperTicks()
-
-def align_subplot_array(axes,xlim=None, ylim=None):
-    """
-    Make all of the axes in the array hae the same limits, turn off unnecessary ticks
-    use plt.subplots() to get an array of axes
-    """
-    #find sensible xlim,ylim
-    if xlim is None:
-        xlim = [np.inf,-np.inf]
-        for ax in axes.flatten():
-            xlim[0] = min(xlim[0],ax.get_xlim()[0])
-            xlim[1] = max(xlim[1],ax.get_xlim()[1])
-    if ylim is None:
-        ylim = [np.inf,-np.inf]
-        for ax in axes.flatten():
-            ylim[0] = min(ylim[0],ax.get_ylim()[0])
-            ylim[1] = max(ylim[1],ax.get_ylim()[1])
-
-    N,M = axes.shape
-    for i,ax in enumerate(axes.flatten()):
-        ax.set_xlim(xlim)
-        ax.set_ylim(ylim)
-        if (i)%M:
-            ax.set_yticks([])
-        else:
-            removeRightTicks(ax)
-        if i<(M*(N-1)):
-            ax.set_xticks([])
-        else:
-            removeUpperTicks(ax)
-
 def x_frame1D(X,plot_limits=None,resolution=None):
     """
     Internal helper function for making plots, returns a set of input values to plot as well as lower and upper limits

GPy/testing/plotting_tests.py

@@ -38,7 +38,7 @@ from nose import SkipTest
 
 try:
     import matplotlib
-    # matplotlib.use('agg')
+    matplotlib.use('agg', warn=False)
 except ImportError:
     # matplotlib not installed
     from nose import SkipTest
@@ -48,6 +48,7 @@ from unittest.case import TestCase
 
 import numpy as np
 import GPy, os
+import logging
 
 from GPy.util.config import config
 from GPy.plotting import change_plotting_library, plotting_library
@@ -98,18 +99,26 @@ def _image_comparison(baseline_images, extensions=['pdf','svg','png'], tol=11, r
     for num, base in zip(plt.get_fignums(), baseline_images):
         for ext in extensions:
             fig = plt.figure(num)
+            try:
                 fig.canvas.draw()
+            except Exception as e:
+                logging.error(base)
+                raise SkipTest(e)
             #fig.axes[0].set_axis_off()
             #fig.set_frameon(False)
             if ext in ['npz']:
                 figdict = flatten_axis(fig)
                 np.savez_compressed(os.path.join(result_dir, "{}.{}".format(base, ext)), **figdict)
+                try:
                     fig.savefig(os.path.join(result_dir, "{}.{}".format(base, 'png')),
                                 transparent=True,
                                 edgecolor='none',
                                 facecolor='none',
                                 #bbox='tight'
                                 )
+                except:
+                    logging.error(base)
+                    raise
             else:
                 fig.savefig(os.path.join(result_dir, "{}.{}".format(base, ext)),
                             transparent=True,

travis_tests.py

@@ -31,7 +31,7 @@
 
 #!/usr/bin/env python
 import matplotlib
-matplotlib.use('agg')
+matplotlib.use('agg', warn=False)
 
 import nose, warnings
 with warnings.catch_warnings():