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)

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)
    umap.plot.points(mapper, labels=iris.target)
    umap.plot.points(mapper, values=iris.data[:, 0])
    umap.plot.points(mapper, theme='fire')
    umap.plot.diagnostic(mapper, diagnostic_type='all')
    umap.plot.diagnostic(mapper, diagnostic_type='neighborhood')
    umap.plot.connectivity(mapper)
    umap.plot.interactive(mapper)
    umap.plot.interactive(mapper, labels=iris.target)
    umap.plot.interactive(mapper, values=iris.data[:, 0])
    umap.plot.interactive(mapper, theme='fire')
    umap.plot._datashade_points(mapper.embedding_)

Add more basic plotting tests
Add more basic plotting tests

  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)
      umap.plot.points(mapper, labels=iris.target)
      umap.plot.points(mapper, values=iris.data[:, 0])
+     umap.plot.points(mapper, theme='fire')
-     umap.plot.diagnostic(mapper, diagnostic_type="all")
+     umap.plot.diagnostic(mapper, diagnostic_type='all')
+     umap.plot.diagnostic(mapper, diagnostic_type='neighborhood')
      umap.plot.connectivity(mapper)
+     umap.plot.interactive(mapper)
+     umap.plot.interactive(mapper, labels=iris.target)
+     umap.plot.interactive(mapper, values=iris.data[:, 0])
+     umap.plot.interactive(mapper, theme='fire')
+     umap.plot._datashade_points(mapper.embedding_)
  
Add more basic plotting tests
## Code Before:
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)

## Instruction:
Add more basic plotting tests
## Code After:
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)
    umap.plot.points(mapper, labels=iris.target)
    umap.plot.points(mapper, values=iris.data[:, 0])
    umap.plot.points(mapper, theme='fire')
    umap.plot.diagnostic(mapper, diagnostic_type='all')
    umap.plot.diagnostic(mapper, diagnostic_type='neighborhood')
    umap.plot.connectivity(mapper)
    umap.plot.interactive(mapper)
    umap.plot.interactive(mapper, labels=iris.target)
    umap.plot.interactive(mapper, values=iris.data[:, 0])
    umap.plot.interactive(mapper, theme='fire')
    umap.plot._datashade_points(mapper.embedding_)
