#481 add tests

nansencenter · Aug 20, 2020 · b5e64d9 · b5e64d9
1 parent 9bb34c0
commit b5e64d9
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/nansat/tests/test_utils.py b/nansat/tests/test_utils.py
@@ -21,13 +21,15 @@
 from mock import patch
 
 try:
-    from matplotlib.colors import hex2color
+    import matplotlib
 except ImportError:
     MATPLOTLIB_IS_INSTALLED = False
 else:
     MATPLOTLIB_IS_INSTALLED = True
+    import matplotlib.pyplot as plt
+    from matplotlib.colors import hex2color
 
-from nansat.utils import get_random_color, parse_time
+from nansat.utils import get_random_color, parse_time, register_colormaps
 from nansat.tests import nansat_test_data as ntd
 
 
@@ -57,3 +59,9 @@ def test_parse_time_incorrect(self):
         dt = parse_time('2016-01-19Z')
 
         self.assertEqual(type(dt), datetime.datetime)
+
+    @unittest.skipUnless(MATPLOTLIB_IS_INSTALLED, 'Matplotlib is required')
+    def test_register_colormaps(self):
+        register_colormaps()
+        self.assertIn('obpg', plt.colormaps())
+        self.assertIn('ak01', plt.colormaps())