update test

Unidata · Sep 5, 2020 · 3376525 · 3376525
1 parent cb6a78e
commit 3376525
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/test/test_cftime.py b/test/test_cftime.py
@@ -18,6 +18,10 @@
                     DatetimeGregorian, DatetimeJulian, DatetimeNoLeap,
                     DatetimeProlepticGregorian, JulianDayFromDate, _parse_date,
                     date2index, date2num, num2date, utime, UNIT_CONVERSION_FACTORS)
+try:
+    from cftime import datetime_base
+except ImportError:
+    datetime_base = datetime
 
 try:
     from datetime import timezone
@@ -830,6 +834,7 @@ def test_tz_naive(self):
         for cal in cftime.DATE_TYPES.keys():
             dt = cftime.datetime(2020, 1, 1, calendar=cal)
             assert(isinstance(dt, cftime.DATE_TYPES[cal]))
+            assert(isinstance(dt, datetime_base))
 
 
 class TestDate2index(unittest.TestCase):