Adjust test for coverage.

Author: anevolbap

causalpy/tests/test_pymc_experiments.py

@@ -10,15 +10,16 @@
 def test_did_summary():
     """Test that the summary stat function returns a string."""
     df = cp.load_data("did")
+    round_to = 2
     result = cp.pymc_experiments.DifferenceInDifferences(
         df,
         formula="y ~ 1 + group*post_treatment",
         time_variable_name="t",
         group_variable_name="group",
         model=cp.pymc_models.LinearRegression(sample_kwargs=sample_kwargs),
     )
-    print(type(result._causal_impact_summary_stat()))
-    assert isinstance(result._causal_impact_summary_stat(), str)
+    print(type(result._causal_impact_summary_stat(round_to)))
+    assert isinstance(result._causal_impact_summary_stat(round_to), str)
 
 
 def test_regression_kink_gradient_change():