def test_load_lymphoma_logrank():
# from https://www.statsdirect.com/help/content/survival_analysis/logrank.htm
df_ = load_lymphoma()
results = stats.multivariate_logrank_test(df_["Time"], df_["Stage_group"],
df_["Censor"])
assert abs(results.test_statistic - 6.70971) < 1e-4
assert abs(results.p_value - 0.0096) < 1e-4
def load_data():
df = load_lymphoma()
df.loc[:, 'Event'] = df.Censor
df = df.drop(['Censor'], axis=1)
return df
