fix: plotting_test

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)
@@ -50,73 +52,73 @@ def gradient_fill(x, percentiles, ax=None, fignum=None, **kwargs):
         kwargs['linewidth'] = 0.5
     if not 'alpha' in kwargs.keys():
         kwargs['alpha'] = 1./(len(percentiles))
-    
+
     # pop where from kwargs
     where = kwargs.pop('where') if 'where' in kwargs else None
     # pop interpolate, which we actually do not do here!
     if 'interpolate' in kwargs: kwargs.pop('interpolate')
-    
+
     def pairwise(inlist):
         l = len(inlist)
         for i in range(int(np.ceil(l/2.))):
             yield inlist[:][i], inlist[:][(l-1)-i]
-    
+
     polycol = []
     for y1, y2 in pairwise(percentiles):
         import matplotlib.mlab as mlab
         # Handle united data, such as dates
         ax._process_unit_info(xdata=x, ydata=y1)
         ax._process_unit_info(ydata=y2)
-    
+
         # Convert the arrays so we can work with them
         from numpy import ma
         x = ma.masked_invalid(ax.convert_xunits(x))
         y1 = ma.masked_invalid(ax.convert_yunits(y1))
         y2 = ma.masked_invalid(ax.convert_yunits(y2))
-    
+
         if y1.ndim == 0:
             y1 = np.ones_like(x) * y1
         if y2.ndim == 0:
             y2 = np.ones_like(x) * y2
-    
+
         if where is None:
             where = np.ones(len(x), np.bool)
         else:
             where = np.asarray(where, np.bool)
-    
+
         if not (x.shape == y1.shape == y2.shape == where.shape):
             raise ValueError("Argument dimensions are incompatible")
-    
+
         mask = reduce(ma.mask_or, [ma.getmask(a) for a in (x, y1, y2)])
         if mask is not ma.nomask:
             where &= ~mask
-        
+
         polys = []
         for ind0, ind1 in mlab.contiguous_regions(where):
             xslice = x[ind0:ind1]
             y1slice = y1[ind0:ind1]
             y2slice = y2[ind0:ind1]
-    
+
             if not len(xslice):
                 continue
-    
+
             N = len(xslice)
             X = np.zeros((2 * N + 2, 2), np.float)
-    
+
             # the purpose of the next two lines is for when y2 is a
             # scalar like 0 and we want the fill to go all the way
             # down to 0 even if none of the y1 sample points do
             start = xslice[0], y2slice[0]
             end = xslice[-1], y2slice[-1]
-    
+
             X[0] = start
             X[N + 1] = end
-    
+
             X[1:N + 1, 0] = xslice
             X[1:N + 1, 1] = y1slice
             X[N + 2:, 0] = xslice[::-1]
             X[N + 2:, 1] = y2slice[::-1]
-    
+
             polys.append(X)
         polycol.extend(polys)
     from matplotlib.collections import PolyCollection
@@ -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