llm/SIA
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix issues in response_parser_test

Browse Source
This commit is contained in:
Niels Geens
2024-11-13 14:33:56 +01:00
parent f248e293de
commit 65478e0b17
1 changed files with 5 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
test/response_parser_test.py
View File

@@ -61,7 +61,7 @@ class ResponseParserTest(unittest.TestCase):
self.assertIsInstance(result, ParseErrorEntry)
self.assertEqual(result.content, xml)
self.assertIn("missing script content", result.error)
self.assertIn("Background entry requires (only) script content", result.error)
def test_repeat_entry(self):
"""Test parsing repeat entry"""
@@ -78,7 +78,7 @@ class ResponseParserTest(unittest.TestCase):
self.assertIsInstance(result, ParseErrorEntry)
self.assertEqual(result.content, xml)
self.assertIn("missing script content", result.error)
self.assertIn("Repeat entry requires (only) script content", result.error)
def test_single_entry(self):
"""Test parsing single shot entry"""
@@ -95,7 +95,7 @@ class ResponseParserTest(unittest.TestCase):
self.assertIsInstance(result, ParseErrorEntry)
self.assertEqual(result.content, xml)
self.assertIn("missing script content", result.error)
self.assertIn("Single entry requires (only) script content", result.error)
def test_reasoning_entry(self):
"""Test parsing reasoning entry"""
@@ -112,7 +112,7 @@ class ResponseParserTest(unittest.TestCase):
self.assertIsInstance(result, ParseErrorEntry)
self.assertEqual(result.content, xml)
self.assertIn("missing content", result.error)
self.assertIn("Reasoning entry requires (only) text content", result.error)
def test_read_stdin_entry(self):
"""Test parsing read stdin entry"""
@@ -137,7 +137,7 @@ class ResponseParserTest(unittest.TestCase):
self.assertIsInstance(result, ParseErrorEntry)
self.assertEqual(result.content, xml)
self.assertIn("missing content", result.error)
self.assertIn("Write stdout entry requires (only) text content", result.error)
def test_unknown_element(self):
"""Test parsing unknown element returns error entry"""
@@ -180,15 +180,6 @@ class ResponseParserTest(unittest.TestCase):
self.assertIsInstance(result, ReasoningEntry)
self.assertTrue(result.content.strip())
def test_script_with_special_characters(self):
"""Test parsing script with special characters"""
script = 'echo "Special chars: \n\t\r\'\"\\"'
xml = f'<single>{script}</single>'
result = self.parser.parse(xml)
self.assertIsInstance(result, SingleEntry)
self.assertEqual(result.script, script)
def test_cdata_content(self):
"""Test parsing content with CDATA sections"""
xml = """