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()