SolowModel

Complexity

Total Complexity

6

Size/Duplication

Total Lines	53
Duplicated Lines	0 %

Importance

Changes	5
Bugs	0	Features	0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	_initial_condition()	0	3	1
A	_rhs_factory()	0	3	1
A	intensive_output()	0	3	1
A	_solow_model()	0	3	1
A	equilibrium_capital()	0	3	1
A	__init__()	0	18	1

import functools

from ... import problems


class SolowModel(problems.IVP):
    """
    Class representing a generic Solow growth model.

    Attributes
    ----------
    equilibrium_capital : function
        Equilibrium value for capital (per unit effective labor).
    intensive_output : function
        Output (per unit effective labor supply).
    params : dict(str: float)
        Dictionary of model parameters.

    """

    def __init__(self, f, k_star, params):
        """
        Initialize an instance of the SolowModel class.

        Parameters
        ----------
        f : function
            Output (per unit effective labor supply).
        k_star : function
            Equilibrium value for capital (per unit effective labor).
        params : dict(str: float)
            Dictionary of model parameters.

        """
        rhs = self._rhs_factory(f)
        self._equilbrium_capital = k_star
        self._intensive_output = f
        super(SolowModel, self).__init__(self._initial_condition, 1, 1, params, rhs)

    @property
    def equilibrium_capital(self):
        return self._equilbrium_capital

    @property
    def intensive_output(self):
        return self._intensive_output

    @staticmethod
    def _initial_condition(t, k, k0, **params):
        return [k - k0]

    @classmethod
    def _solow_model(cls, t, k, f, delta, g, n, s, **params):
        return [s * f(k, **params) - (g + n + delta) * k]

    @classmethod
    def _rhs_factory(cls, f):
        return functools.partial(cls._solow_model, f=f)