Fixed test_drawdown_overlaps, by now testing properly for NaT.

quantopian · Jul 15, 2020 · 4b901f6 · 4b901f6
1 parent ecc8308
commit 4b901f6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyfolio/tests/test_timeseries.py b/pyfolio/tests/test_timeseries.py
@@ -182,7 +182,7 @@ def test_drawdown_overlaps(self):
                          spy_drawdowns['Peak date'].shift(-1)))[:-1]
         self.assertGreater(len(pairs), 0)
         for recovery, peak in pairs:
-            if recovery != pd.NaT:
+            if not pd.isnull(recovery):
                 self.assertLessEqual(recovery, peak)
 
     @parameterized.expand([