示例#1
0
def test_estimators(estimator, check, request):
    # Common tests for estimator instances
    with ignore_warnings(category=(FutureWarning, ConvergenceWarning,
                                   UserWarning, FutureWarning)):
        _set_checking_parameters(estimator)

        xfail_checks = _safe_tags(estimator, '_xfail_test')
        check_name = _set_check_estimator_ids(check)
        if xfail_checks:
            if check_name in xfail_checks:
                msg = xfail_checks[check_name]
                request.applymarker(pytest.mark.xfail(reason=msg))
        check(estimator)
def test_set_check_estimator_ids(val, expected):
    assert _set_check_estimator_ids(val) == expected