test_miscplot.py 963 B

1234567891011121314151617181920212223242526272829303132333435
  1. import nose.tools as nt
  2. import matplotlib.pyplot as plt
  3. from .. import miscplot as misc
  4. from ..palettes import color_palette
  5. from ..utils import _network
  6. class TestPalPlot(object):
  7. """Test the function that visualizes a color palette."""
  8. def test_palplot_size(self):
  9. pal4 = color_palette("husl", 4)
  10. misc.palplot(pal4)
  11. size4 = plt.gcf().get_size_inches()
  12. nt.assert_equal(tuple(size4), (4, 1))
  13. pal5 = color_palette("husl", 5)
  14. misc.palplot(pal5)
  15. size5 = plt.gcf().get_size_inches()
  16. nt.assert_equal(tuple(size5), (5, 1))
  17. palbig = color_palette("husl", 3)
  18. misc.palplot(palbig, 2)
  19. sizebig = plt.gcf().get_size_inches()
  20. nt.assert_equal(tuple(sizebig), (6, 2))
  21. class TestDogPlot(object):
  22. @_network(url="https://github.com/mwaskom/seaborn-data")
  23. def test_dogplot(self):
  24. misc.dogplot()
  25. ax = plt.gca()
  26. assert len(ax.images) == 1