Commit af58e0b6 authored by Adam Paquette's avatar Adam Paquette
Browse files

Changed in method cluster attribute check from computing the cluster, to... 

Changed in method cluster attribute check from computing the cluster, to throwing an attribute error.
parent f1540480

autocnet/vis/graph_view.py

+5 −3
Original line number Diff line number Diff line
@@ -106,10 +106,10 @@ def plot_node(node, ax=None, clean_keys=[], index_mask=None, **kwargs): 


    return ax





def plot_edge_decomposition(edge, ax=None, clean_keys=[], image_space=100,

                            scatter_kwargs={}, line_kwargs={}, image_kwargs={}):





    if ax is None:

        ax = plt.gca()
@@ -175,6 +175,8 @@ def plot_edge_decomposition(edge, ax=None, clean_keys=[], image_space=100, 
        ax.plot((l[0][0], l[1][0]), (l[0][1], l[1][1]), color=color, **line_kwargs)



    return ax





def plot_edge(edge, ax=None, clean_keys=[], image_space=100,

              scatter_kwargs={}, line_kwargs={}, image_kwargs={}):

    """
@@ -273,7 +275,7 @@ def plot_edge(edge, ax=None, clean_keys=[], image_space=100, 
    return ax





def cluster_plot(graph, ax=None, cmap='Spectral'):

def cluster_plot(graph, ax=None, cmap='Spectral'):  # pragma: no cover

    """

    Parameters

    ----------
@@ -296,7 +298,7 @@ def cluster_plot(graph, ax=None, cmap='Spectral'): 
        ax = plt.gca()



    if not hasattr(graph, 'clusters'):

        graph.compute_clusters()

        raise AttributeError('Clusters have not been computed.')



    cmap = matplotlib.cm.get_cmap(cmap)