make sure option is passed down num2date

rcomer · rcomer · commit 97bd7f75bf05 · 2021-05-10T19:19:29.000+01:00
diff --git a/cf_units/__init__.py b/cf_units/__init__.py
@@ -2033,4 +2033,6 @@ def num2date(self, time_value, only_use_cftime_datetimes=False):
         cdf_utime = self.utime(
             only_use_cftime_datetimes=only_use_cftime_datetimes)
 
-        return _num2date_to_nearest_second(time_value, cdf_utime)
+        return _num2date_to_nearest_second(
+            time_value, cdf_utime,
+            only_use_cftime_datetimes=only_use_cftime_datetimes)
diff --git a/cf_units/tests/test_unit.py b/cf_units/tests/test_unit.py
@@ -29,6 +29,7 @@
 from operator import truediv
 
 import numpy as np
+import cftime
 
 import cf_units as unit
 from cf_units import suppress_errors
@@ -262,7 +263,7 @@ def test_add_float_offset(self):
     def test_not_numerical_offset(self):
         u = Unit('meter')
         with self.assertRaisesRegex(TypeError,
-                                   'unsupported operand type'):
+                                    'unsupported operand type'):
             operator.add(u, 'not_a_number')
 
     def test_unit_unknown(self):
@@ -990,6 +991,13 @@ def test_num2date(self):
         self.assertEqual(str(res), '2010-11-02 13:00:00')
         self.assertIsInstance(res, datetime.datetime)
 
+    def test_num2date_cftime_type(self):
+        u = Unit('hours since 2010-11-02 12:00:00',
+                 calendar=unit.CALENDAR_STANDARD)
+        res = u.num2date(1, only_use_cftime_datetimes=True)
+        self.assertEqual(str(res), '2010-11-02 13:00:00')
+        self.assertIsInstance(res, cftime.DatetimeGregorian)
+
     def test_date2num(self):
         u = Unit('hours since 2010-11-02 12:00:00',
                  calendar=unit.CALENDAR_STANDARD)
