some new features

.venv/lib/python3.12/site-packages/prophet/tests/test_utilities.py

@@ -0,0 +1,27 @@
+# Copyright (c) Facebook, Inc. and its affiliates.
+
+# This source code is licensed under the MIT license found in the
+# LICENSE file in the root directory of this source tree.
+
+import numpy as np
+
+from prophet import Prophet
+from prophet.utilities import regressor_coefficients
+
+
+class TestUtilities:
+    def test_regressor_coefficients(self, daily_univariate_ts, backend):
+        m = Prophet(stan_backend=backend)
+        df = daily_univariate_ts.copy()
+        np.random.seed(123)
+        df["regr1"] = np.random.normal(size=df.shape[0])
+        df["regr2"] = np.random.normal(size=df.shape[0])
+        m.add_regressor("regr1", mode="additive")
+        m.add_regressor("regr2", mode="multiplicative")
+        m.fit(df)
+
+        coefs = regressor_coefficients(m)
+        assert coefs.shape == (2, 6)
+        # No MCMC sampling, so lower and upper should be the same as mean
+        assert np.array_equal(coefs["coef_lower"].values, coefs["coef"].values)
+        assert np.array_equal(coefs["coef_upper"].values, coefs["coef"].values)