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