def test_infers_classification_for_object_target(self):
y = pd.Series(['high', 'low'])
assert 'classification' == infer_ml_task(y)
def test_infers_regression_for_float_target(self):
y = pd.Series([1.0, 1.5, 1.7])
assert 'regression' == infer_ml_task(y)
def test_infers_classification_for_integer_target(self):
y = pd.Series([1, 2, 3])
assert 'classification' == infer_ml_task(y)
def test_infers_classification_for_boolean_target(self):
y = pd.Series([True, False, False])
assert 'classification' == infer_ml_task(y)
def test_infers_regression_for_float_target(self):
y = pd.Series([1.0, 1.5, 1.7])
assert 'regression' == infer_ml_task(y)
def test_infers_classification_for_object_target(self):
y = pd.Series(['high', 'low'])
assert 'classification' == infer_ml_task(y)
def test_infers_classification_for_boolean_target(self):
y = pd.Series([True, False, False])
assert 'classification' == infer_ml_task(y)
def test_infers_classification_for_integer_target(self):
y = pd.Series([1, 2, 3])
assert 'classification' == infer_ml_task(y)
def test_infers_classification_for_object_target(self):
y = pd.Series(["high", "low"])
assert "classification" == infer_ml_task(y)