cumulated_study.py

import numpy as np

from experiment.meteo_france_SCM_models.study.abstract_study import AbstractStudy
from experiment.meteo_france_SCM_models.study.abstract_variable import AbstractVariable


class CumulatedStudy(AbstractStudy):
    def __init__(self, variable_class: type, nb_consecutive_days: int = 1, *args, **kwargs):
        assert nb_consecutive_days in [1, 3, 5, 7]
        super().__init__(variable_class, *args, **kwargs)
        self.nb_consecutive_days = nb_consecutive_days

    def instantiate_variable_object(self, variable_array) -> AbstractVariable:
        return self.variable_class(variable_array, self.nb_consecutive_days)

    @property
    def variable_name(self):
        return super().variable_name + ' cumulated over {} day(s)'.format(self.nb_consecutive_days)