climate-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From jarifibra...@apache.org
Subject [2/2] climate git commit: Add tests for temporal_resolution function
Date Mon, 13 Jun 2016 19:51:23 GMT
Add tests for temporal_resolution function


Project: http://git-wip-us.apache.org/repos/asf/climate/repo
Commit: http://git-wip-us.apache.org/repos/asf/climate/commit/5ecaa8e6
Tree: http://git-wip-us.apache.org/repos/asf/climate/tree/5ecaa8e6
Diff: http://git-wip-us.apache.org/repos/asf/climate/diff/5ecaa8e6

Branch: refs/heads/master
Commit: 5ecaa8e6adedbd07cdefc6f841587953f7343679
Parents: aaa3bc5
Author: Ibrahim <jarifibrahim@gmail.com>
Authored: Tue May 31 16:30:38 2016 +0530
Committer: Ibrahim <jarifibrahim@gmail.com>
Committed: Tue Jun 14 00:51:58 2016 +0530

----------------------------------------------------------------------
 ocw/tests/test_dataset.py | 75 ++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 73 insertions(+), 2 deletions(-)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/climate/blob/5ecaa8e6/ocw/tests/test_dataset.py
----------------------------------------------------------------------
diff --git a/ocw/tests/test_dataset.py b/ocw/tests/test_dataset.py
index 8a1bad5..d30fbcc 100644
--- a/ocw/tests/test_dataset.py
+++ b/ocw/tests/test_dataset.py
@@ -142,9 +142,52 @@ class TestDatasetFunctions(unittest.TestCase):
     def test_spatial_resolution(self):
         self.assertEqual(self.test_dataset.spatial_resolution(), (2, 2))
 
-    def test_temporal_resolution(self):
+    def test_temporal_resolution_monthly(self):
         self.assertEqual(self.test_dataset.temporal_resolution(), 'monthly')
 
+    def test_temporal_resolution_daily(self):
+        self.time = np.array([dt.datetime(2000, 3, x) for x in range(1, 31)])
+        flat_array = np.array(range(750))
+        self.value = flat_array.reshape(30, 5, 5)
+        self.test_dataset = Dataset(self.lat, self.lon, self.time,
+                                    self.value, self.variable)
+        self.assertEqual(self.test_dataset.temporal_resolution(), 'daily')
+
+    def test_temporal_resolution_yearly(self):
+        self.time = np.array([dt.datetime(x, 6, 1) for x in range(2000, 2015)])
+        flat_array = np.array(range(375))
+        self.value = flat_array.reshape(15, 5, 5)
+        self.test_dataset = Dataset(self.lat, self.lon, self.time,
+                                    self.value, self.variable)
+        self.assertEqual(self.test_dataset.temporal_resolution(), 'yearly')
+
+    def test_str_(self):
+        dataset = self.test_dataset
+        lat_min, lat_max, lon_min, lon_max = dataset.spatial_boundaries()
+        start, end = dataset.time_range()
+        lat_range = "({}, {})".format(lat_min, lon_min)
+        lon_range = "({}, {})".format(lon_min, lon_min)
+        time_range = "({}, {})".format(start, end)
+
+        formatted_repr = (
+            "<Dataset - name: {}, "
+            "lat-range: {}, "
+            "lon-range: {}, "
+            "time_range: {}, "
+            "var: {}, "
+            "units: {}>"
+        )
+
+        output = formatted_repr.format(
+            dataset.name if dataset.name != "" else None,
+            lat_range,
+            lon_range,
+            time_range,
+            dataset.variable,
+            dataset.units
+        )
+        self.assertEqual(str(self.test_dataset), output)
+
 
 class TestBounds(unittest.TestCase):
     def setUp(self):
@@ -152,7 +195,8 @@ class TestBounds(unittest.TestCase):
                              -160, 160,               # Lons
                              dt.datetime(2000, 1, 1),  # Start time
                              dt.datetime(2002, 1, 1))  # End time
-
+        self.another_bounds = Bounds(-80, 80,                # Lats
+                                     -160, 160)
     # Latitude tests
     def test_inverted_min_max_lat(self):
         with self.assertRaises(ValueError):
@@ -219,5 +263,32 @@ class TestBounds(unittest.TestCase):
         with self.assertRaises(ValueError):
             self.bounds.end = "This is not a date time object"
 
+    # Start tests
+    def test_none_value_start(self):
+        self.assertEqual(self.another_bounds.start, None)
+
+    # End tests
+    def test_none_value_end(self):
+        self.assertEqual(self.another_bounds.end, None)
+
+    def test__str__(self):
+        lat_range = "({}, {})".format(self.bounds.lat_min, self.bounds.lat_max)
+        lon_range = "({}, {})".format(self.bounds.lon_min, self.bounds.lon_max)
+        time_range = "({}, {})".format(self.bounds.start, self.bounds.end)
+
+        formatted_repr = (
+            "<Bounds - "
+            "lat-range: {}, "
+            "lon-range: {}, "
+            "time_range: {}> "
+        )
+
+        output = formatted_repr.format(
+            lat_range,
+            lon_range,
+            time_range,
+        )
+        self.assertEqual(str(self.bounds), output)
+
 if __name__ == '__main__':
     unittest.main()


Mime
View raw message