edit makefile and add test to test.py

Makefile

@@ -3,7 +3,7 @@ install:
 		pip install -r requirements.txt
 
 test:
-	python -m pytest app.py
+	python -m pytest test.py
 	#--nbval notebook/EDA.ipynb notebook/MODEL_TRAINING.ipynb
 
 format:

test.py

@@ -1,7 +1,23 @@
 from src.components.data_ingestion import DataIngestion
+from src.components.data_transformation import DataTransformation
 
 def test_data_ingestion_output():
     obj = DataIngestion()
     train_data, test_data = obj.initiate_data_ingestion()
     assert train_data == "artifacts/train.csv"
-    assert test_data == "artifacts/test.csv"
+    assert test_data == "artifacts/test.csv"
+
+
+def test_data_transformation_output():
+    obj = DataTransformation()
+    preprocessor = obj.get_data_transformer_object()
+    assert preprocessor is not None
+
+    train_path = "artifacts/train.csv"
+    test_path = "artifacts/test.csv"
+
+    train_arr, test_arr, preprocessor_path = obj.initiate_data_transformation(train_path, test_path)
+
+    assert train_arr is not None
+    assert test_arr is not None
+    assert preprocessor_path is not None