def validate_method(op: 'lale.operators.IndividualOp', schema_name: str):
    """Check whether the operator has the given method schema.

    Parameters
    ----------
    op: lale.operators.IndividualOp
        Operator whose methods to check.

    schema_name: 'input_fit' or 'input_predict' or 'input_predict_proba' or 'input_transform' 'output_predict' or 'output_predict_proba' or 'output_transform'
        Name of schema to check.

    Raises
    ------
    AssertionError
        The operator does not have the given schema.
    """
    if op._impl.__module__.startswith('lale'):
        assert schema_name in op._schemas['properties']
    else:
        method_name = ''
        if schema_name.startswith('input_'):
            method_name = schema_name[len('input_'):]
        elif schema_name.startswith('output_'):
            method_name = schema_name[len('output_'):]
        if method_name:
            assert hasattr(op._impl, method_name)
