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"))