def test_ajf_plotting_no_confidence_intervals(self, block): E = [0, 1, 1, 2, 2, 0] T = [1, 2, 3, 4, 5, 6] ajf = AalenJohansenFitter(calculate_variance=False).fit(T, E, event_of_interest=1) ajf.plot(ci_show=False) self.plt.title("test_ajf_plotting_no_confidence_intervals") self.plt.show(block=block) return
def test_ajf_plotting_with_add_count_at_risk(self, block): E = [0, 1, 1, 2, 2, 0] T = [1, 2, 3, 4, 5, 6] ajf = AalenJohansenFitter().fit(T, E, event_of_interest=1) ajf.plot(at_risk_counts=True) self.plt.title("test_ajf_plotting_with_add_count_at_risk") self.plt.show(block=block) return
def test_ajf_plotting(self, block): E = [0, 1, 1, 2, 2, 0] T = [1, 2, 3, 4, 5, 6] ajf = AalenJohansenFitter().fit(T, E, event_of_interest=1) ajf.plot() self.plt.title("test_ajf_plotting") self.plt.show(block=block) return