Update Turnbull test

This commit is contained in:
RunasSudo 2023-12-31 18:34:40 +11:00
parent 8e30c41cc4
commit f42c152019
Signed by: RunasSudo
GPG Key ID: 7234E476BF21C61A

View File

@ -28,7 +28,7 @@ def test_turnbull_simple():
'Time_R': [2.5, 2, 4, 6] 'Time_R': [2.5, 2, 4, 6]
}) })
xpoints, ypoints = yli.survival.calc_survfunc_turnbull(df['Time_L'], df['Time_R']) xpoints, ypoints, _, _ = yli.survival.calc_survfunc_turnbull(df['Time_L'], df['Time_R'], ci=False)
assert xpoints[0] == 0 assert xpoints[0] == 0
assert ypoints[0] == 1 assert ypoints[0] == 1