fix test

CamDavidsonPilon · CamDavidsonPilon · commit 04b2081e9bc2 · 2024-10-29T06:30:54.000-04:00
diff --git a/lifelines/tests/test_estimation.py b/lifelines/tests/test_estimation.py
@@ -3320,7 +3320,7 @@ def test_strata_estimation_is_same_if_using_trivial_strata(self, rossi):
 
         assert_frame_equal(
             cph.summary.loc[[("beta_", "Intercept"), ("phi1_", "Intercept")]].reset_index(drop=True),
-            trivial_strata_cph.summary.loc[[("beta_", "Intercept"), ("sa_phi1_", "Intercept")]].reset_index(drop=True),
+            trivial_strata_cph.summary.loc[[("beta_", "Intercept"), ("s('a',)_phi1_", "Intercept")]].reset_index(drop=True),
             atol=0.05,
         )
 

Original file line number	Diff line number	Diff line change
`@@ -3320,7 +3320,7 @@ def test_strata_estimation_is_same_if_using_trivial_strata(self, rossi):`
`3320`	`3320`
`3321`	`3321`	`assert_frame_equal(`
`3322`	`3322`	`cph.summary.loc[[("beta_", "Intercept"), ("phi1_", "Intercept")]].reset_index(drop=True),`
`3323`		`- trivial_strata_cph.summary.loc[[("beta_", "Intercept"), ("sa_phi1_", "Intercept")]].reset_index(drop=True),`
	`3323`	`+ trivial_strata_cph.summary.loc[[("beta_", "Intercept"), ("s('a',)_phi1_", "Intercept")]].reset_index(drop=True),`
`3324`	`3324`	`atol=0.05,`
`3325`	`3325`	`)`
`3326`	`3326`