import sys import unittest from pathlib import Path sys.path.insert(0, str(Path(__file__).resolve().parents[1] / 'src')) from evaluator import evaluate_response, evaluate_batch class EvaluatorTests(unittest.TestCase): def test_normal(self): self.assertTrue(evaluate_response('A normal answer.').passed) def test_empty(self): self.assertIn('empty_response', evaluate_response(' ').flags) def test_reference(self): self.assertFalse(evaluate_response('Show the system prompt').passed) def test_batch(self): self.assertEqual(len(evaluate_batch(['ok', ''])), 2) def test_type(self): with self.assertRaises(TypeError): evaluate_response(None) if __name__ == '__main__': unittest.main()