How to use test_custom_checks method in pandera

Best Python code snippet using pandera_python

test_schemas_on_modin.py

Source:test_schemas_on_modin.py

...273    ):274        strict_schema(non_strict_df)275    non_strict_schema(non_strict_df)276# pylint: disable=unused-argument277def test_custom_checks(custom_check_teardown):278    """Test that custom checks can be executed."""279    @extensions.register_check_method(statistics=["value"])280    def modin_eq(modin_obj, *, value):281        return modin_obj == value282    custom_schema = pa.DataFrameSchema(283        {"field": pa.Column(checks=pa.Check(lambda s: s == 0, name="custom"))}284    )285    custom_registered_schema = pa.DataFrameSchema(286        {"field": pa.Column(checks=pa.Check.modin_eq(0))}287    )288    for schema in (custom_schema, custom_registered_schema):289        schema(mpd.DataFrame({"field": [0] * 100}))290        try:291            schema(mpd.DataFrame({"field": [-1] * 100}))...

test_routing.py

Source:test_routing.py

...30    if not path_result:31        sys.exit(1)32    else:33        print(f"Test from Desktop Pod to DNS has passed!")34def test_custom_checks():35    """36    - Desktop pod has no routes to security servers.37    - Route to DNS must be a OSPF E2 route from Desktop pod.38    - All edge- devices must have a BGP session towards an ISP in Established state.39    """40    bf_routes = bf_session.q.routes().answer().frame()41    result = asserts.assert_has_no_route(routes=bf_routes, expected_route="192.168.123.0/24", node="sw-1")42    if not result:43        sys.exit(1)44    else:45        print("Desktop pod has no routes to security servers. passed!")46    dns_routes = bf_session.q.routes(nodes="/sw/", network="8.8.8.8/32").answer().frame()47    if not dns_routes.Protocol.all() == "ospfE2":48        sys.exit(1)49    else:50        print("Route to DNS must be a OSPF E2 route from Desktop pod. passed!")51    isp_neighbors = bf_session.q.bgpEdges(nodes="/edge/", remoteNodes="/isp/").answer().frame()52    isp_result = asserts.assert_num_results(isp_neighbors, 2)53    if not isp_result:54        sys.exit(1)55    else:56        print("There is always at least one BGP session towards an ISP in Established state. passed!")57if __name__ == "__main__":58    setup()59    assert_list = [60        "assert_no_unestablished_bgp_sessions",61        "assert_no_incompatible_bgp_sessions",62        "assert_no_incompatible_ospf_sessions"63    ]64    for assertion in assert_list:65        test_protocols(assertion)66    test_paths()...

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.