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)
