Test your AI Agents with the all-new Agent to Agent Testing Platform.Learn More

How to use multiindex_strategy method in pandera

Best Python code snippet using pandera_python

test_strategies.py

Source:test_strategies.py

...545@hypothesis.given(st.data())546@hypothesis.settings(547    suppress_health_check=[hypothesis.HealthCheck.too_slow],548)549def test_multiindex_strategy(data) -> None:550    """Test MultiIndex schema component strategy."""551    data_type = pa.Float()552    multiindex = pa.MultiIndex(553        indexes=[554            pa.Index(data_type, unique=True, name="level_0"),555            pa.Index(data_type, nullable=True),556            pa.Index(data_type),557        ]558    )559    strat = multiindex.strategy(size=10)560    example = data.draw(strat)561    for i in range(example.nlevels):562        actual_data_type = pandas_engine.Engine.dtype(563            example.get_level_values(i).dtype564        )565        assert data_type.check(actual_data_type)566    with pytest.raises(pa.errors.BaseStrategyOnlyError):567        strategies.multiindex_strategy(568            data_type, strategies.pandas_dtype_strategy(data_type)569        )570def test_multiindex_example() -> None:571    """572    Test MultiIndex schema component example method generates examples that573    pass.574    """575    data_type = pa.Float()576    multiindex = pa.MultiIndex(577        indexes=[578            pa.Index(data_type, unique=True, name="level_0"),579            pa.Index(data_type, nullable=True),580            pa.Index(data_type),581        ]...

schema_components.py

Source:schema_components.py

...654    # NOTE: remove these ignore statements as part of655    # https://github.com/pandera-dev/pandera/issues/403656    # pylint: disable=arguments-differ657    def strategy(self, *, size=None):  # type: ignore658        return st.multiindex_strategy(indexes=self.indexes, size=size)659    # NOTE: remove these ignore statements as part of660    # https://github.com/pandera-dev/pandera/issues/403661    # pylint: disable=arguments-differ662    def example(self, size=None) -> pd.MultiIndex:  # type: ignore663        # pylint: disable=import-outside-toplevel,cyclic-import,import-error664        import hypothesis665        with warnings.catch_warnings():666            warnings.simplefilter(667                "ignore",668                category=hypothesis.errors.NonInteractiveExampleWarning,669            )670            return self.strategy(size=size).example()671    def __repr__(self):672        return (...

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.