run testes on assessments predictions pipeline

This commit is contained in:
2024-09-12 00:01:03 +00:00
parent 007c0d81dc
commit 24823432b3
14 changed files with 254 additions and 16 deletions
@@ -0,0 +1,18 @@
import unittest
from src.pipeline.data_preprocessor import DataPreprocessor
import os
class TestDataPreprocessor(unittest.TestCase):
def setUp(self):
self.dp = DataPreprocessor(
input_path="/root/ds_erp_ai/data/raw/dummy_assessment_data.csv",
company_id="company_id"
)
def test_run(self):
res = self.dp.run()
self.assertIsNotNone(res) # Check that the result is not None
self.assertTrue(os.path.exists(res)) # Check that the output file exists
if __name__ == '__main__':
unittest.main()