def test_check_y_converts_string_to_number():
s = pd.Series(["0", "1", "2", "3", "4"])
assert_series_equal(check_y(s, y_numeric=True), s.astype("float"))
def test_check_y_raises_inf_error():
s = pd.Series([0, np.inf, 2, 3, 4])
with pytest.raises(ValueError):
check_y(s)
def test_check_y_converts_np_array():
a1D = np.array([1, 2, 3, 4])
s = pd.Series(a1D)
assert_series_equal(check_y(a1D), s)
def test_check_y_raises_none_error():
with pytest.raises(ValueError):
check_y(None)
def test_check_y_returns_series():
s = pd.Series([0, 1, 2, 3, 4])
assert_series_equal(check_y(s), s)