def test_kmf_with_interval_censoring_plotting(self, block):
kmf = KaplanMeierFitter()
left, right = load_diabetes()["left"], load_diabetes()["right"]
kmf.fit_interval_censoring(left, right)
kmf.plot(color="r")
self.plt.show(block=block)
return
def test_max_lower_bound_less_than_min_upper_bound():
# https://github.com/CamDavidsonPilon/lifelines/issues/1151
import numpy as np
import pandas as pd
from lifelines import KaplanMeierFitter
# Data
np.random.seed(1)
left0 = np.random.normal(loc=60, scale=2, size=20)
add_time = np.random.normal(loc=100, scale=2, size=10)
right1 = left0[0:10] + add_time
right0 = right1.tolist() + [np.inf] * 10
# KaplanMeier
model = KaplanMeierFitter()
model.fit_interval_censoring(lower_bound=left0, upper_bound=right0)
