                                   7.00000000e+00, 8.00000000e+00,  8.333333e+00])
        cls.model_json = model_to_json(cls.model)

    def test_model_save_and_load(self):
        prophet_model = ProphetModel(self.model_json, 1, "d", "ds")

        with mlflow.start_run() as run:
            mlflow_prophet_log_model(prophet_model)
        # Load the saved model from mlflow
        run_id = run.info.run_id
        prophet_model = mlflow.pyfunc.load_model(f"runs:/{run_id}/model")