Skip to content
Prev Previous commit
Next Next commit
update formatting
  • Loading branch information
madel0093 committed Oct 29, 2020
commit 4cf640f8db6c4af2f1c49f8b8c129b11622375e7
29 changes: 22 additions & 7 deletions tests/test_datasets/test_dataset_functions.py
Original file line number Diff line number Diff line change
Expand Up @@ -1419,11 +1419,19 @@ def test_data_fork(self):

@pytest.mark.parametrize(
"default_target_attribute,row_id_attribute,ignore_attribute",
[("wrong", None, None), (None, "wrong", None), (None, None, "wrong"),
("wrong,sunny", None, None),(None, None, "wrong,sunny"),
(["wrong","sunny"], None, None), (None, None, ["wrong","sunny"])],
[
("wrong", None, None),
(None, "wrong", None),
(None, None, "wrong"),
("wrong,sunny", None, None),
(None, None, "wrong,sunny"),
(["wrong", "sunny"], None, None),
(None, None, ["wrong", "sunny"]),
],
)
def test_invalid_attribute_validations(default_target_attribute, row_id_attribute, ignore_attribute):
def test_invalid_attribute_validations(
default_target_attribute, row_id_attribute, ignore_attribute
):
data = [
["a", "sunny", 85.0, 85.0, "FALSE", "no"],
["b", "sunny", 80.0, 90.0, "TRUE", "no"],
Expand Down Expand Up @@ -1467,11 +1475,18 @@ def test_invalid_attribute_validations(default_target_attribute, row_id_attribut
paper_url=paper_url,
)


@pytest.mark.parametrize(
"default_target_attribute,row_id_attribute,ignore_attribute",
[("outlook", None, None), (None, "outlook", None), (None, None, "outlook"),
("outlook,windy", None, None), (None, None, "outlook,windy"),
(["outlook","windy"], None, None), (None, None, ["outlook","windy"])],
[
("outlook", None, None),
(None, "outlook", None),
(None, None, "outlook"),
("outlook,windy", None, None),
(None, None, "outlook,windy"),
(["outlook", "windy"], None, None),
(None, None, ["outlook", "windy"]),
],
)
def test_valid_attribute_validations(default_target_attribute, row_id_attribute, ignore_attribute):
data = [
Expand Down