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