You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@climate.apache.org by bo...@apache.org on 2014/05/27 23:19:21 UTC
[2/3] git commit: CLIMATE-439 unittest for calc_climatology_year in
utils.py
CLIMATE-439 unittest for calc_climatology_year in utils.py
Project: http://git-wip-us.apache.org/repos/asf/climate/repo
Commit: http://git-wip-us.apache.org/repos/asf/climate/commit/74e400c6
Tree: http://git-wip-us.apache.org/repos/asf/climate/tree/74e400c6
Diff: http://git-wip-us.apache.org/repos/asf/climate/diff/74e400c6
Branch: refs/heads/master
Commit: 74e400c614573638523e471095d07dd23e1504e7
Parents: 8a7f1d4
Author: Maziyar Boustani <ma...@yahoo.com>
Authored: Tue May 27 13:37:46 2014 -0700
Committer: Maziyar Boustani <ma...@yahoo.com>
Committed: Tue May 27 13:37:46 2014 -0700
----------------------------------------------------------------------
ocw/tests/test_utils.py | 29 +++++++++++++++++++++++++++--
1 file changed, 27 insertions(+), 2 deletions(-)
----------------------------------------------------------------------
http://git-wip-us.apache.org/repos/asf/climate/blob/74e400c6/ocw/tests/test_utils.py
----------------------------------------------------------------------
diff --git a/ocw/tests/test_utils.py b/ocw/tests/test_utils.py
index f775b76..6044d2f 100644
--- a/ocw/tests/test_utils.py
+++ b/ocw/tests/test_utils.py
@@ -159,7 +159,7 @@ class TestReshapeMonthlyToAnnually(unittest.TestCase):
self.value, self.variable)
def test_reshape_full_year(self):
- new_values = self.value.reshape(2,12,5,5)
+ new_values = self.value.reshape(2, 12, 5, 5)
np.testing.assert_array_equal(
utils.reshape_monthly_to_annually(self.test_dataset), new_values)
@@ -169,7 +169,32 @@ class TestReshapeMonthlyToAnnually(unittest.TestCase):
value = flat_array.reshape(26, 5, 5)
bad_dataset = Dataset(self.lat, self.lon, new_time, value, self.variable)
- self.assertRaises(ValueError,utils.reshape_monthly_to_annually,bad_dataset)
+ self.assertRaises(ValueError, utils.reshape_monthly_to_annually, bad_dataset)
+
+class TestCalcClimatologyYear(unittest.TestCase):
+ ''' Testing function 'calc_climatology_year' from ocw.utils.py '''
+
+ def setUp(self):
+ self.lat = np.array([10, 12, 14, 16, 18])
+ self.lon = np.array([100, 102, 104, 106, 108])
+ self.time = np.array([datetime.datetime(2000, 1, 1) + relativedelta(months = x) for x in range(24)])
+ flat_array = np.array(range(600))
+ self.value = flat_array.reshape(24, 5, 5)
+ self.variable = 'prec'
+ self.test_dataset = Dataset(self.lat, self.lon, self.time,
+ self.value, self.variable)
+
+ def test_annually_mean(self):
+ annually_mean = np.append(np.arange(137.5, 162.5, 1), np.arange(437.5, 462.5, 1))
+ annually_mean.shape = (2, 5, 5)
+ np.testing.assert_array_equal(
+ utils.calc_climatology_year(self.test_dataset)[0], annually_mean)
+
+ def test_total_mean(self):
+ total_mean = np.arange(287.5, 312.5, 1)
+ total_mean.shape = (5, 5)
+ np.testing.assert_array_equal(
+ utils.calc_climatology_year(self.test_dataset)[1], total_mean)
if __name__ == '__main__':
unittest.main()