from nose.tools import assert_less
from nose.tools import assert_greater_equal
import os.path
import numpy as np

from nose import SkipTest

from sklearn import datasets

import umap
import umap.plot

np.random.seed(42)
iris = datasets.load_iris()

mapper = umap.UMAP(n_epochs=100).fit(iris.data)


def test_plot_runs_at_all():
    umap.plot.points(mapper, labels=iris.target)
    umap.plot.points(mapper, values=iris.data[:, 0])
    umap.plot.diagnostic(mapper, diagnostic_type="all")
    umap.plot.connectivity(mapper)

@@ -321,16 +321,45 @@
 _all():%0A
+    umap.plot.points(mapper)%0A
     umap
@@ -399,16 +399,16 @@
 target)%0A
-
     umap
@@ -452,16 +452,59 @@
 %5B:, 0%5D)%0A
+    umap.plot.points(mapper, theme='fire')%0A
     umap
@@ -548,46 +548,356 @@
 ype=
-%22all%22)%0A    umap.plot.connectivity(mapper
+'all')%0A    umap.plot.diagnostic(mapper, diagnostic_type='neighborhood')%0A    umap.plot.connectivity(mapper)%0A    umap.plot.interactive(mapper)%0A    umap.plot.interactive(mapper, labels=iris.target)%0A    umap.plot.interactive(mapper, values=iris.data%5B:, 0%5D)%0A    umap.plot.interactive(mapper, theme='fire')%0A    umap.plot._datashade_points(mapper.embedding_
 )%0A

Add more basic plotting tests