def check()

in jetstream/statistics.py [0:0]

• 12 lines of code
• 3 McCabe index (conditional complexity)

    def check(self, attribute, value):
        if value is not None:
            covariate_period = parser_metric.AnalysisPeriod(value["period"])
            preenrollment_periods = [
                parser_metric.AnalysisPeriod.PREENROLLMENT_WEEK,
                parser_metric.AnalysisPeriod.PREENROLLMENT_DAYS_28,
            ]
            if covariate_period not in preenrollment_periods:
                raise ValueError(
                    "Covariate adjustment must be done using a pre-treatment analysis "
                    f"period (one of: {[p.value for p in preenrollment_periods]})"
                )