#91 sd-50 test_figure: add tests

nansat/figure.py

@@ -100,6 +100,8 @@ class Figure(object):
 
     _cmapName = 'jet'
 
+    # instance attributes
+    array = None
     def __init__(self, nparray, **kwargs):
         ''' Set attributes
 

nansat/tests/test_figure.py

@@ -12,12 +12,24 @@
 #               under the terms of GNU General Public License, v.3
 #               http://www.gnu.org/licenses/gpl-3.0.html
 #------------------------------------------------------------------------------
+from __future__ import absolute_import, unicode_literals
+
 import unittest
 import warnings
 import os
 import sys
 import glob
 import datetime
+if sys.version_info.major == 2:
+    from mock import patch, PropertyMock, Mock, MagicMock, DEFAULT
+else:
+    from unittest.mock import patch, PropertyMock, Mock, MagicMock, DEFAULT
+try:
+    import Image
+    import ImageDraw
+    import ImageFont
+except:
+    from PIL import Image, ImageDraw, ImageFont
 
 import gdal
 import numpy as np
@@ -123,5 +135,24 @@ def test_get_tick_index_from_grid(self):
         n.logger.error(str(lonTicksIdx))
         n.logger.error(str(latTicksIdx))
 
+    @patch.object(Figure, '__init__', return_value=None)
+    def test_apply_logarithm(self, mock1):
+        f = Figure()
+        f.array = np.ones((1,2,2))*0.1
+        f.apply_logarithm()
+        self.assertTrue(np.allclose(np.ones((1,2,2))*0.31622777, f.array))
+
+    @patch.object(Figure, '__init__', return_value=None)
+    def test_make_transparent_color(self, mock1):
+        f = Figure()
+
+        img_array = np.array([[1.,2.],[3.,4.]])
+        f.pilImg = Image.fromarray(img_array)
+        f.reprojMask = np.zeros((2,2)) == 1
+        f.transparency = [1,1,1]
+        f._make_transparent_color()
+        self.assertTrue((np.array(f.pilImg)[:,:,3] == np.array([[0,255],[255,255]])).all())
+
+
 if __name__ == "__main__":
     unittest.main()

nansat/tests/test_vrt.py

@@ -10,7 +10,6 @@
 #               under the terms of GNU General Public License, v.3
 #               http://www.gnu.org/licenses/gpl-3.0.html
 # ------------------------------------------------------------------------------
-
 from __future__ import absolute_import, unicode_literals
 import unittest
 import logging