run testes on assessments predictions pipeline
This commit is contained in:
@@ -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()
|
||||
Reference in New Issue
Block a user