def test_default_feature_list(self):
dataset = DataSet(pd.DataFrame([[4, 1, 2], [5, 9, 8], [4, 3, 6]]))
assert_that(dataset.feature_list(), contains(0, 1, 2))
def test_feature_list(self):
dataset = DataSet(pd.DataFrame([[4, 1, 2], [5, 9, 8], [4, 3, 6]],
columns=["a", "b", "c"]))
assert_that(dataset.feature_list(), contains("a", "b", "c"))
