18 lines
567 B
Python
18 lines
567 B
Python
|
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()
|