def test_rmst_plot_with_single_model(self, block):
waltons = load_waltons()
kmf = KaplanMeierFitter().fit(waltons["T"], waltons["E"])
rmst_plot(kmf, t=40.0)
self.plt.title("test_rmst_plot_with_single_model")
self.plt.show(block=block)
def test_rmst_plot_with_two_model(self, block):
waltons = load_waltons()
ix = waltons["group"] == "control"
kmf_con = KaplanMeierFitter().fit(waltons.loc[ix]["T"], waltons.loc[ix]["E"], label="control")
kmf_exp = KaplanMeierFitter().fit(waltons.loc[~ix]["T"], waltons.loc[~ix]["E"], label="exp")
rmst_plot(kmf_con, model2=kmf_exp, t=40.0)
self.plt.title("test_rmst_plot_with_two_model")
self.plt.show(block=block)
