Allow timedelta to be converted to an ordinalf

dstansby · dstansby · commit 80a7cbdfcf56 · 2017-08-29T18:28:35.000+01:00
diff --git a/lib/matplotlib/dates.py b/lib/matplotlib/dates.py
@@ -239,6 +239,17 @@ def _to_ordinalf(dt):
 _to_ordinalf_np_vectorized = np.vectorize(_to_ordinalf)
 
 
+def _tdelta_to_ordinalf(tdelta):
+    """
+    Convert :mod:`timedelta` to total days. Return value is a :func:`float`
+    """
+    return tdelta.total_seconds() / SEC_PER_DAY
+
+
+# a version of _tdelta_to_ordinalf that can operate on numpy arrays
+_tdelta_to_ordinalf_np_vectorized = np.vectorize(_tdelta_to_ordinalf)
+
+
 def _from_ordinalf(x, tz=None):
     """
     Convert Gregorian float of the date, preserving hours, minutes,
diff --git a/lib/matplotlib/tests/test_dates.py b/lib/matplotlib/tests/test_dates.py
@@ -474,3 +474,10 @@ def test_tz_utc():
 def test_num2timedelta(x, tdelta):
     dt = mdates.num2timedelta(x)
     assert dt == tdelta
+
+
+def test_timedelta_ordinalf():
+    # Check that timedeltas can be converted to ordinalfs
+    dt = datetime.timedelta(seconds=60)
+    ordinalf = mdates._tdelta_to_ordinalf(dt)
+    assert ordinalf == 1 / (24 * 60)
