Updating integration test

JeanVanDyk · JeanVanDyk · commit 5c4eb13077e6 · 2025-07-09T12:43:12.000+02:00
diff --git a/causalpy/tests/test_integration_pymc_examples.py b/causalpy/tests/test_integration_pymc_examples.py
@@ -431,7 +431,7 @@ def test_its_no_treatment_time():
             model=cp.pymc_models.LinearRegression(sample_kwargs=sample_kwargs),
         )
     assert (
-        "If treatment_time is None, provided model must have a 'set_time_range' method"
+        "If treatment_time is a tuple, provided model must have a 'set_time_range' method"
         in str(exc_info.value)
     )
 

Original file line number	Diff line number	Diff line change
`@@ -431,7 +431,7 @@ def test_its_no_treatment_time():`
`431`	`431`	`model=cp.pymc_models.LinearRegression(sample_kwargs=sample_kwargs),`
`432`	`432`	`)`
`433`	`433`	`assert (`
`434`		`- "If treatment_time is None, provided model must have a 'set_time_range' method"`
	`434`	`+ "If treatment_time is a tuple, provided model must have a 'set_time_range' method"`
`435`	`435`	`in str(exc_info.value)`
`436`	`436`	`)`
`437`	`437`