testing data validation pipeline

Author: GoJo-Rika

test.py

@@ -1,8 +1,11 @@
+## Data Validation Testing
 import sys
 
 from network_security.components.data_ingestion import DataIngestion
+from network_security.components.data_validation import DataValidation
 from network_security.entity.config_entity import (
     DataIngestionConfig,
+    DataValidationConfig,
     TrainingPipelineConfig,
 )
 from network_security.exception.exception import NetworkSecurityException
@@ -17,6 +20,48 @@
         data_ingestion = DataIngestion(data_ingestion_config=data_ingestion_config)
         logging.info("Initiated data ingestion")
         data_ingestion_artifact = data_ingestion.initiate_data_ingestion()
+        logging.info("Data Initiation Completed")
         print(data_ingestion_artifact)
+
+        data_validation_config = DataValidationConfig(
+            training_pipeline_config=training_pipeline_config,
+        )
+        data_validation = DataValidation(
+            data_ingestion_artifact=data_ingestion_artifact,
+            data_validation_config=data_validation_config,
+        )
+        logging.info("Initiated data validation")
+        data_validation_artifact = data_validation.initiate_data_validation()
+        logging.info("Data Validation Completed")
+        print(data_validation_artifact)
+
     except Exception as e:
         raise NetworkSecurityException(e, sys)
+
+
+## =============================================================
+
+## Data Ingestion Testing
+
+# import sys
+
+# from network_security.components.data_ingestion import DataIngestion
+# from network_security.entity.config_entity import (
+#     DataIngestionConfig,
+#     TrainingPipelineConfig,
+# )
+# from network_security.exception.exception import NetworkSecurityException
+# from network_security.logging.logger import logging
+
+# if __name__ == "__main__":
+#     try:
+#         training_pipeline_config = TrainingPipelineConfig()
+#         data_ingestion_config = DataIngestionConfig(
+#             training_pipeline_config=training_pipeline_config,
+#         )
+#         data_ingestion = DataIngestion(data_ingestion_config=data_ingestion_config)
+#         logging.info("Initiated data ingestion")
+#         data_ingestion_artifact = data_ingestion.initiate_data_ingestion()
+#         print(data_ingestion_artifact)
+#     except Exception as e:
+#         raise NetworkSecurityException(e, sys)