def test_ftrl_set_individual():
ft = Ftrl()
ft.alpha = tparams.alpha
ft.beta = tparams.beta
ft.lambda1 = tparams.lambda1
ft.lambda2 = tparams.lambda2
ft.d = tparams.d
ft.nepochs = tparams.nepochs
ft.inter = tparams.inter
assert ft.params == tparams
def test_ftrl_set_wrong_d_value():
ft = Ftrl()
with pytest.raises(ValueError) as e:
ft.d = 0
assert ("Value should be positive: 0" == str(e.value))
def test_ftrl_set_wrong_d_type():
ft = Ftrl()
with pytest.raises(TypeError) as e:
ft.d = "0"
assert ("Expected an integer, instead got <class 'str'>" == str(e.value))
