Skip to content
GitLab
Select Git revision
  • master default
Find file BlameHistoryPermalink
cumulated_study.py 794 bytes 






1


2


3


4


5


6


7


8


9


10


11


12


13


14


15


16


17


18






from experiment.meteo_france_data.scm_models_data.abstract_study import AbstractStudy
from experiment.meteo_france_data.scm_models_data.abstract_variable import AbstractVariable

NB_DAYS = [1, 3, 5, 7]


class CumulatedStudy(AbstractStudy):
    def __init__(self, variable_class: type, nb_consecutive_days: int = 3, *args, **kwargs):
        assert nb_consecutive_days in NB_DAYS
        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)