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)