testing data ingestion pipeline

Author: GoJo-Rika

test.py

@@ -0,0 +1,22 @@
+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)