Add unit tests

Add unit tests for critical areas, giving priority to following.

1. Preprocessing
2. Model training
3. Classifying

L’issue mentionne le test de preprocessing, model training et classifying. Cherche des fichiers de test existants ou un répertoire tests/ pour comprendre le pattern de test du projet. Examine la codebase pour localiser les modules de preprocessing, training et classification. Écris des tests unitaires pour ces zones, en t’assurant qu’ils s’exécutent avec le test runner du projet (probablement pytest). Vérifie que les tests passent et couvrent la logique critique.

Rédigé par le modèle d'indexation à partir du texte de l'issue.