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
