Example #1
0
def test_Estimator__init__():
    d = Wrapped(LogisticRegression(C=1000))
    assert d._name == from_sklearn(clf1)._name

    with pytest.raises(ValueError):
        Wrapped(clf1, name='foo')

    with pytest.raises(TypeError):
        Wrapped("not an estimator")
Example #2
0
def test_dir():
    d = Wrapped(clf1)
    attrs = dir(d)
    assert 'C' in attrs
Example #3
0
def test_getattr():
    d = Wrapped(clf1)
    assert d.C == clf1.C
    with pytest.raises(AttributeError):
        d.not_a_real_parameter
Example #4
0
def test_setattr():
    d = Wrapped(clf1)
    with pytest.raises(AttributeError):
        d.C = 10
Example #5
0
def test__estimator_type():
    d = Wrapped(clf1)
    assert d._estimator_type == clf1._estimator_type
Example #6
0
def test_clone():
    d = Wrapped(clf1)
    d2 = clone(d)
    assert d.get_params() == d2.get_params()
    assert d._est is not d2._est
Example #7
0
def test_repr():
    d = Wrapped(clf1)
    res = repr(d)
    assert res.startswith('Wrapped')
Example #8
0
def test_tokenize_Averaged():
    c1 = Averaged(sgdc_1)
    c2 = Averaged(sgdc_2)
    assert tokenize(c1) == tokenize(c1)
    assert tokenize(c1) != tokenize(c2)
    assert tokenize(c1) != tokenize(Wrapped(sgdc_1))
Example #9
0
def test_tokenize_Chained():
    c1 = Chained(sgd1)
    c2 = Chained(sgd2)
    assert tokenize(c1) == tokenize(c1)
    assert tokenize(c1) != tokenize(c2)
    assert tokenize(c1) != tokenize(Wrapped(sgd1))