def test_feature_fulldata_apply(self): data = [[1, 0, 1], [0, 2, 0], [1, 0, 1]] columns = ["col1", "col2", "col3"] data = pd.DataFrame(data, columns=columns) feature = Data(x_train=data, split=False, report_name="test") feature.apply(lambda x: x["col1"] > 0, "new_col") validate = "new_col" in feature.x_train.columns self.assertTrue(validate)
def test_feature_splitdata_apply(self): data = [["py", 0, 1], ["auto", 2, 0], ["ml", 0, 1]] columns = ["col1", "col2", "col3"] data = pd.DataFrame(data, columns=columns) feature = Data(x_train=data, test_split_percentage=0.33, report_name="test") feature.apply(lambda x: x["col1"], "new_col") validate = ("new_col" in feature.x_train.columns and "new_col" in feature.x_test.columns) self.assertTrue(validate)