llm-finetuned-model / tests /test_preprocessing.py
promptsbyesha's picture
Upload folder using huggingface_hub
72f4d4d verified
raw
history blame contribute delete
397 Bytes
import unittest
from src.preprocessing import preprocess_data
class TestPreprocessing(unittest.TestCase):
def test_sample(self):
sample_data = ["Hello World!"]
processed = preprocess_data(sample_data)
self.assertIsInstance(processed, list)
self.assertTrue(all(isinstance(s, str) for s in processed))
if __name__ == "__main__":
unittest.main()