add variable_unit_conversion function tests
Project: http://git-wip-us.apache.org/repos/asf/climate/repo Commit: http://git-wip-us.apache.org/repos/asf/climate/commit/fb53c09c Tree: http://git-wip-us.apache.org/repos/asf/climate/tree/fb53c09c Diff: http://git-wip-us.apache.org/repos/asf/climate/diff/fb53c09c Branch: refs/heads/master Commit: fb53c09c54eefb27cac1c5303c1d7295afd5f48b Parents: dd93d7f Author: Ibrahim <jarifibra...@gmail.com> Authored: Sun May 29 23:47:01 2016 +0530 Committer: Ibrahim <jarifibra...@gmail.com> Committed: Tue Jun 14 01:06:10 2016 +0530 ---------------------------------------------------------------------- ocw/tests/test_dataset_processor.py | 62 ++++++++++++++++++++++++++++++++ 1 file changed, 62 insertions(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/climate/blob/fb53c09c/ocw/tests/test_dataset_processor.py ---------------------------------------------------------------------- diff --git a/ocw/tests/test_dataset_processor.py b/ocw/tests/test_dataset_processor.py index 1271995..c186e4b 100644 --- a/ocw/tests/test_dataset_processor.py +++ b/ocw/tests/test_dataset_processor.py @@ -95,6 +95,68 @@ class TestTemporalRebinWithTimeIndex(unittest.TestCase): np.testing.assert_array_equal(self.ten_year_dataset.lons, dataset.lons) +class TestVariableUnitConversion(unittest.TestCase): + def setUp(self): + self.ten_year_dataset = ten_year_monthly_dataset() + self.ten_year_dataset.variable = 'temp' + self.ten_year_dataset.units = 'celsius' + + def test_returned_variable_unit_celsius(self): + ''' Tests returned dataset unit if original dataset unit is celcius ''' + dp.variable_unit_conversion(self.ten_year_dataset) + self.assertEqual(self.ten_year_dataset.units, 'K') + + def test_returned_variable_unit_kelvin(self): + ''' Tests returned dataset unit if original dataset unit is kelvin ''' + self.ten_year_dataset.units = 'K' + another_dataset = dp.variable_unit_conversion(self.ten_year_dataset) + self.assertEqual(another_dataset.units, self.ten_year_dataset.units) + + def test_temp_unit_conversion(self): + ''' Tests returned dataset temp values ''' + self.ten_year_dataset.values = np.ones([ + len(self.ten_year_dataset.times), + len(self.ten_year_dataset.lats), + len(self.ten_year_dataset.lons)]) + values = self.ten_year_dataset.values + 273.15 + dp.variable_unit_conversion(self.ten_year_dataset) + np.testing.assert_array_equal(self.ten_year_dataset.values, values) + + def test_returned_variable_unit_swe(self): + ''' Tests returned dataset unit if original dataset unit is swe ''' + self.ten_year_dataset.variable = 'swe' + self.ten_year_dataset.units = 'm' + dp.variable_unit_conversion(self.ten_year_dataset) + self.assertEqual(self.ten_year_dataset.variable, 'swe') + self.assertEqual(self.ten_year_dataset.units, 'km') + + def test_returned_variable_unit_pr(self): + ''' + Tests returned dataset unit if original dataset unit is kgm^-2s^-1 + ''' + self.ten_year_dataset.variable = 'pr' + self.ten_year_dataset.units = 'kg m-2 s-1' + dp.variable_unit_conversion(self.ten_year_dataset) + self.assertEqual(self.ten_year_dataset.variable, 'pr') + self.assertEqual(self.ten_year_dataset.units, 'mm/day') + + def test_water_flux_unit_conversion_swe(self): + ''' Tests variable values in returned dataset ''' + self.ten_year_dataset.variable = 'swe' + self.ten_year_dataset.units = 'm' + values = self.ten_year_dataset.values + 999 + dp.variable_unit_conversion(self.ten_year_dataset) + np.testing.assert_array_equal(self.ten_year_dataset.values, values) + + def test_water_flux_unit_conversion_pr(self): + ''' Tests variable values in returned dataset ''' + self.ten_year_dataset.variable = 'pr' + self.ten_year_dataset.units = 'kg m-2 s-1' + values = self.ten_year_dataset.values + 86399 + dp.variable_unit_conversion(self.ten_year_dataset) + np.testing.assert_array_equal(self.ten_year_dataset.values, values) + + class TestEnsemble(unittest.TestCase): def test_unequal_dataset_shapes(self): self.ten_year_dataset = ten_year_monthly_dataset()
