test_model.py

import pytest
from inference import predict
from utils import preprocess_text

def test_preprocessing():
    text = "  Hello   World  "
    processed = preprocess_text(text)
    assert processed == "Hello World"

def test_prediction():
    # Test with simple input
    result = predict("Test input")
    assert isinstance(result, list)
    print("Basic tests passed!")

if __name__ == "__main__":
    test_preprocessing()
    test_prediction()