Source code for macrostat.models.GL06SIMEX.scenarios

"""
Scenarios class for the Godley-Lavoie 2006 SIMEX model.
"""

__author__ = ["Karl Naumann-Woleske"]
__credits__ = ["Karl Naumann-Woleske"]
__license__ = "MIT"
__maintainer__ = ["Karl Naumann-Woleske"]

import logging

from macrostat.core.scenarios import Scenarios
from macrostat.models.GL06SIMEX.parameters import ParametersGL06SIMEX

logger = logging.getLogger(__name__)




class ScenariosGL06SIMEX(Scenarios):
    """Scenarios class for the Godley-Lavoie 2006 SIMEX model."""

    version = "GL06SIMEX"

    def __init__(
        self,
        scenario_info: dict | None = None,
        parameters: ParametersGL06SIMEX | None = None,
        *args,
        **kwargs,
    ):
        """Initialize the scenarios of the Godley-Lavoie 2006 SIMEX model."""

        if parameters is None:
            parameters = ParametersGL06SIMEX()

        super().__init__(
            scenario_info=scenario_info,
            parameters=parameters,
            *args,
            **kwargs,
        )



    def get_default_scenario_values(self):
        """Return the default scenario values."""

        sc = {
            "GovernmentDemand": 20,
            "WageRate": 1,
        }

        for k in self.parameters.values.keys():
            sc[f"{k.replace('.', '_')}_add"] = 0.0

        return sc