diff --git a/rdagent/components/coder/data_science/feature/eval_tests/feature_test.txt b/rdagent/components/coder/data_science/feature/eval_tests/feature_test.txt
index 726c258a..a99d75ae 100644
--- a/rdagent/components/coder/data_science/feature/eval_tests/feature_test.txt
+++ b/rdagent/components/coder/data_science/feature/eval_tests/feature_test.txt
@@ -38,6 +38,10 @@ def get_column_list(data):
     return data.columns.tolist() if isinstance(data, pd.DataFrame) else None
 
 
+assert X is not None, "The feature engineering function returned None for X."
+assert y is not None, "The feature engineering function returned None for y."
+assert X_test is not None, "The feature engineering function returned None for X_test."
+
 assert get_length(X_test) == get_length(
     test_ids
 ), f"Mismatch in length of test images and test IDs: X_test ({get_length(X_test)}) and test_ids ({get_length(test_ids)})"
diff --git a/rdagent/components/coder/data_science/raw_data_loader/eval_tests/data_loader_test.txt b/rdagent/components/coder/data_science/raw_data_loader/eval_tests/data_loader_test.txt
index f9d4e757..750dfe11 100644
--- a/rdagent/components/coder/data_science/raw_data_loader/eval_tests/data_loader_test.txt
+++ b/rdagent/components/coder/data_science/raw_data_loader/eval_tests/data_loader_test.txt
@@ -26,6 +26,10 @@ def get_width(data):
 def get_column_list(data):
     return data.columns.tolist() if isinstance(data, pd.DataFrame) else None
 
+assert X is not None, "Training data (X) is None."
+assert y is not None, "Training labels (y) are None."
+assert X_test is not None, "Test data (X_test) is None."
+assert test_ids is not None, "Test IDs (test_ids) are None."
 
 assert get_length(X_test) == get_length(
     test_ids