added tests for two new util functions

Author: ValentinGebhart

climada/util/test/test_checker.py

@@ -109,6 +109,15 @@ def test_prune_csr_matrix(self):
         np.testing.assert_array_equal(matrix.data, [3])
         self.assertEqual(matrix.nnz, 1)
 
+    def test_convert_frequency_unit_to_time_unit(self):
+        # test frequency unit to time unit conversion
+        frequency_units = ['1/year', '1/y', '1/month', '1/week', '', 'unknown']
+        time_units = ['years', 'years', 'months', 'weeks', 'years', 'years']
+        np.testing.assert_equal(
+            time_units, [u_check.convert_frequency_unit_to_time_unit(frequency_unit)
+                         for frequency_unit in frequency_units]
+        )
+
 # Execute Tests
 if __name__ == "__main__":
     TESTS = unittest.TestLoader().loadTestsFromTestCase(TestChecks)

climada/util/test/test_interpolation.py

@@ -22,7 +22,7 @@
 import unittest
 import numpy as np
 
-from climada.util.interpolation import interpolate_ev, stepfunction_ev, group_frequency
+from climada.util.interpolation import interpolate_ev, stepfunction_ev, group_frequency, preprocess_and_interpolate_ev
 
 
 class TestFitMethods(unittest.TestCase):
@@ -192,6 +192,41 @@ def test_frequency_group(self):
         with self.assertRaises(ValueError):
             group_frequency(frequency, intensity[::-1])
 
+    def test_preprocess_and_interpolate_ev(self):
+        """Test wrapper function"""
+        frequency = np.array([.1, .9])
+        values = np.array([100., 10.])
+        test_frequency = np.array([.01, .55, 10.])
+        test_values = np.array([1., 55., 1000.])
+
+        # test interpolation
+        np.testing.assert_allclose(
+            [np.nan, 55., np.nan],
+            preprocess_and_interpolate_ev(test_frequency, None, frequency, values)
+        )
+        np.testing.assert_allclose(
+            [np.nan, .55, np.nan],
+            preprocess_and_interpolate_ev(None, test_values, frequency, values)
+        )
+
+        # test extrapolation with constants
+        np.testing.assert_allclose(
+            [100. , 55., 0.],
+            preprocess_and_interpolate_ev(test_frequency, None, frequency, values,
+                                          method='extrapolate_constant', y_asymptotic=0.)
+        )
+        np.testing.assert_allclose(
+            [1., .55, np.nan],
+            preprocess_and_interpolate_ev(None, test_values, frequency, values,
+                                          method='extrapolate_constant')
+        )
+
+        # test error raising
+        with self.assertRaises(ValueError):
+            preprocess_and_interpolate_ev(test_frequency, test_values, frequency, values)
+        with self.assertRaises(ValueError):
+            preprocess_and_interpolate_ev(None, None, frequency, values)
+
 # Execute Tests
 if __name__ == "__main__":
     TESTS = unittest.TestLoader().loadTestsFromTestCase(TestFitMethods)