tests/test_metrics.py

import unittest
from evaluations.evaluate_models import evaluate_model

class TestMetrics(unittest.TestCase):

    def test_evaluate_model(self):
        model = train_full_finetune_model(train_data, val_data)
        results = evaluate_model(model, val_data)
        self.assertIn('accuracy', results)
        self.assertIn('f1_score', results)

if __name__ == '__main__':
    unittest.main()