def test_weibull_aft_plot_covariate_groups(self, block):
df = load_rossi()
aft = WeibullAFTFitter()
aft.fit(df, "week", "arrest")
aft.plot_covariate_groups("age", [10, 50, 80])
self.plt.tight_layout()
self.plt.title("test_weibull_aft_plot_covariate_groups")
self.plt.show(block=block)
def test_weibull_aft_plot_covariate_groups_with_multiple_columns(self, block):
df = load_rossi()
aft = WeibullAFTFitter()
aft.fit(df, "week", "arrest")
aft.plot_covariate_groups(["age", "prio"], [[10, 0], [50, 10], [80, 50]])
self.plt.tight_layout()
self.plt.title("test_weibull_aft_plot_covariate_groups_with_multiple_columns")
self.plt.show(block=block)
