add test for issue #198

Unidata · Sep 25, 2020 · d5e8279 · d5e8279
1 parent 46d4f67
commit d5e8279
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/test/test_cftime.py b/test/test_cftime.py
@@ -819,6 +819,17 @@ def test_tz_naive(self):
         cal = 'proleptic_gregorian'
         dt2 = num2date(date2num(dt1, units, cal), units, cal)
         assert(dt1 == dt2)
+# issue #198 - cftime.datetime creates calendar specific datetimes that
+# support addition/subtraction of timedeltas.
+        dt = cftime.datetime(2020, 1, 1, calendar='')
+        assert(isinstance(dt, cftime.datetime))
+        dt = cftime.datetime(2020, 1, 1, calendar="julian")
+        dt += timedelta(hours=1)
+        assert(str(dt) == '2020-01-01 01:00:00')
+        assert(isinstance(dt, cftime.DatetimeJulian))
+        for cal in cftime.DATE_TYPES.keys():
+            dt = cftime.datetime(2020, 1, 1, calendar=cal)
+            assert(isinstance(dt, cftime.DATE_TYPES[cal]))
 
 
 class TestDate2index(unittest.TestCase):