def _validate_coef_prior()

in orbit/template/ktr.py [0:0]

• 25 lines of code
• 7 McCabe index (conditional complexity)

    def _validate_coef_prior(coef_prior_list):
        for test_dict in coef_prior_list:
            if set(test_dict.keys()) != set(
                [
                    KTRTimePointPriorKeys.NAME.value,
                    KTRTimePointPriorKeys.PRIOR_START_TP_IDX.value,
                    KTRTimePointPriorKeys.PRIOR_END_TP_IDX.value,
                    KTRTimePointPriorKeys.PRIOR_MEAN.value,
                    KTRTimePointPriorKeys.PRIOR_SD.value,
                    KTRTimePointPriorKeys.PRIOR_REGRESSOR_COL.value,
                ]
            ):
                raise IllegalArgument("wrong key name in inserted prior dict")
            len_insert_prior = list()
            for key, val in test_dict.items():
                if key in [
                    KTRTimePointPriorKeys.PRIOR_MEAN.value,
                    KTRTimePointPriorKeys.PRIOR_SD.value,
                    KTRTimePointPriorKeys.PRIOR_REGRESSOR_COL.value,
                ]:
                    len_insert_prior.append(len(val))
            if not all(
                len_insert == len_insert_prior[0] for len_insert in len_insert_prior
            ):
                raise IllegalArgument("wrong dimension length in inserted prior dict")