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

How to use test_check_decorator_coercion method in pandera

Best Python code snippet using pandera_python

test_decorators.py

Source:test_decorators.py

...166    with pytest.raises(167        IndexError, match=r"^error in check_input decorator of function"168    ):169        test_incorrect_check_input_index(pd.DataFrame({"column1": [1, 2, 3]}))170def test_check_decorator_coercion() -> None:171    """Test that check decorators correctly coerce input/output data."""172    in_schema = DataFrameSchema({"column1": Column(int, coerce=True)})173    out_schema = DataFrameSchema({"column2": Column(float, coerce=True)})174    @check_input(in_schema)175    @check_output(out_schema)176    def test_func_io(df):177        return df.assign(column2=10)178    @check_input(in_schema)179    @check_output(out_schema, obj_getter=1)180    def test_func_out_tuple_obj_getter(df):181        return None, df.assign(column2=10)182    @check_input(in_schema)183    @check_output(out_schema, obj_getter=1)184    def test_func_out_list_obj_getter(df):...

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.