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

Fix superfluous CDATA tokens

Browse Source
This commit is contained in:
geens
2024-11-11 14:23:08 +01:00
parent 3a1d8a449c
commit 59f482bc0e
17 changed files with 102 additions and 324 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
test/parse_error_entry_test.py
View File

@@ -48,12 +48,12 @@ class ParseErrorEntryTest(unittest.TestCase):
# Verify error element
error_elem = element.find("error")
self.assertIsNotNone(error_elem)
self.assertEqual(error_elem.text, f"<![CDATA[{error}]]>")
self.assertEqual(error_elem.text, f"{error}")
# Verify content element
content_elem = element.find("content")
self.assertIsNotNone(content_elem)
self.assertEqual(content_elem.text, f"<![CDATA[{content}]]>")
self.assertEqual(content_elem.text, f"{content}")
def test_special_characters(self):
"""Test handling content and errors with special characters"""
@@ -62,13 +62,13 @@ class ParseErrorEntryTest(unittest.TestCase):
entry = ParseErrorEntry(content, error, self.test_id, self.test_timestamp)
element = entry.generate_context()
self.assertEqual(element.find("content").text, f"<![CDATA[{content}]]>")
self.assertEqual(element.find("error").text, f"<![CDATA[{error}]]>")
self.assertEqual(element.find("content").text, f"{content}")
self.assertEqual(element.find("error").text, f"{error}")
def test_empty_content_and_error(self):
"""Test handling empty content and error messages"""
entry = ParseErrorEntry("", "", self.test_id, self.test_timestamp)
element = entry.generate_context()
self.assertEqual(element.find("content").text, "<![CDATA[]]>")
self.assertEqual(element.find("error").text, "<![CDATA[]]>")
self.assertEqual(element.find("content").text, "")
self.assertEqual(element.find("error").text, "")