Unlock 30% off on Manual Testing Annual Plans this Holiday Season.

Copied to Clipboard!

How to use register_check_wrapper method in pandera

Best Python code snippet using pandera_python

extensions.py

Source:extensions.py

...79            supported_types=supported_types,80            check_type=check_type,81            strategy=strategy,82        )83    def register_check_wrapper(check_fn: Callable):84        """Register a function as a :class:`~pandera.checks.Check` method."""85        if hasattr(Check, check_fn.__name__):86            raise ValueError(87                f"method with name '{check_fn.__name__}' already defined. "88                "Check methods must have a unique method name."89            )90        @wraps(check_fn)91        def check_fn_wrapper(validate_obj, **kwargs):92            """Wrapper for check_fn to validate inputs."""93            return check_fn(validate_obj, **kwargs)94        def validate_check_kwargs(check_kwargs):95            msg = (96                f"'{check_fn.__name__} has check_type={check_type}. "97                "Providing the following arguments will have no effect: "98                "{}. Remove these arguments to avoid this warning."99            )100            no_effect_args = {101                CheckType.ELEMENT_WISE: ["element_wise", "groupby", "groups"],102                CheckType.VECTORIZED: ["element_wise", "groupby", "groups"],103                CheckType.GROUPBY: ["element_wise"],104            }[check_type]105            if any(arg in check_kwargs for arg in no_effect_args):106                warnings.warn(msg.format(no_effect_args))107                for arg in no_effect_args:108                    check_kwargs.pop(arg, None)109            if check_type is CheckType.ELEMENT_WISE:110                check_kwargs["element_wise"] = True111            return check_kwargs112        @register_check_statistics(statistics)113        def check_method(cls, *args, **kwargs):114            """Wrapper function that serves as the Check method."""115            stats, check_kwargs = {}, {}116            if args:117                stats = dict(zip(statistics, args))118            for k, v in kwargs.items():119                if k in statistics:120                    stats[k] = v121                else:122                    check_kwargs[k] = v123            return cls(124                partial(check_fn_wrapper, **stats),125                name=check_fn.__name__,126                **validate_check_kwargs(check_kwargs),127            )128        if strategy is not None:129            check_method = st.register_check_strategy(strategy)(check_method)130        Check.REGISTERED_CUSTOM_CHECKS[check_fn.__name__] = partial(131            check_method, Check132        )...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.