| import unittestfrom evaluations.evaluate_models import evaluate_modelclass 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() | |