fix test

CamDavidsonPilon · Aug 28, 2017 · c80a512 · c80a512
1 parent 10efc6a
commit c80a512
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_estimation.py b/tests/test_estimation.py
@@ -441,7 +441,7 @@ def test_kmf_does_not_drop_to_zero_if_last_point_is_censored(self):
         E[np.argmax(T)] = 0
         kmf = KaplanMeierFitter()
         kmf.fit(T, E)
-        assert T.survival_function_[-1] > 0
+        assert kmf.survival_function_['KM_estimate'].iloc[-1] > 0
 
 
 class TestNelsonAalenFitter():