def test_check_y_not_int_not_float(wage_X_y, wage_gam):
"""y must be int or float, or we should get a value error"""
X, y = wage_X_y
y_str = ['hi'] * len(y)
with pytest.raises(ValueError):
check_y(y_str, wage_gam.link, wage_gam.distribution)
def test_check_y_not_int_not_float(wage_X_y, wage_gam):
"""y must be int or float, or we should get a value error"""
X, y = wage_X_y
y_str = ['hi'] * len(y)
try:
check_y(y_str, wage_gam.link, wage_gam.distribution)
assert False
except ValueError:
check_y(y, wage_gam.link, wage_gam.distribution)
assert (True)
def test_check_y_not_in_domain_link(default_X_y, default_gam):
"""if you give labels outide of the links domain, check_y will raise an error"""
X, y = default_X_y
gam = default_gam
with pytest.raises(ValueError):
check_y(y + .1,
default_gam.link,
default_gam.distribution,
verbose=False)
def test_check_y_not_min_samples(wage_X_y, wage_gam):
"""check_y expects a minimum number of samples"""
X, y = wage_X_y
with pytest.raises(ValueError):
check_y(y,
wage_gam.link,
wage_gam.distribution,
min_samples=len(y) + 1,
verbose=False)
def test_check_y_not_in_doamin_link(default, default_gam):
"""if you give labels outide of the links domain, check_y will raise an error"""
X, y = default
gam = default_gam
try:
check_y(y + .1, default_gam.link, default_gam.distribution)
assert False
except ValueError:
check_y(y, default_gam.link, default_gam.distribution)
assert True
def test_check_y_not_min_samples(wage, wage_gam):
"""check_y expects a minimum number of samples"""
X, y = wage
try:
check_y(y,
wage_gam.link,
wage_gam.distribution,
min_samples=len(y) + 1)
assert False
except ValueError:
check_y(y, wage_gam.link, wage_gam.distribution, min_samples=len(y))
assert True
def test_check_y_casts_to_numerical(wage_X_y, wage_gam):
"""check_y will try to cast data to numerical types"""
X, y = wage_X_y
y = y.astype('object')
y = check_y(y, wage_gam.link, wage_gam.distribution)
assert y.dtype == 'float'
