def test_coxtv_plotting(self, block):
df = load_stanford_heart_transplants()
ctv = CoxTimeVaryingFitter()
ctv.fit(df, id_col="id", event_col="event")
ctv.plot(fmt="o")
self.plt.title("test_coxtv_plotting")
self.plt.show(block=block)
def test_coxtv_plotting_with_subset_of_columns(self, block):
df = load_stanford_heart_transplants()
ctv = CoxTimeVaryingFitter()
ctv.fit(df, id_col="id", event_col="event")
ctv.plot(columns=["age", "year"])
self.plt.title("test_coxtv_plotting_with_subset_of_columns")
self.plt.show(block=block)