Fix pretty printing

This commit is contained in:
2024-11-02 15:22:32 +01:00
parent a58ee40c2c
commit 96d14fc46d
6 changed files with 213 additions and 33 deletions

View File

@@ -145,7 +145,6 @@ class BaseAgentTest(unittest.TestCase):
self.assertEqual(len(list(root)), 0)
def test_compile_context_with_entries(self):
"""Test context compilation with working memory entries."""
# Add test entry to working memory
entry = ReasoningEntry("test reasoning", "test-id", self.test_timestamp)
self.agent._working_memory.add_entry(entry)
@@ -162,7 +161,8 @@ class BaseAgentTest(unittest.TestCase):
reasoning_elem = root.find("reasoning")
self.assertIsNotNone(reasoning_elem)
self.assertEqual(reasoning_elem.get("id"), "test-id")
self.assertEqual(reasoning_elem.text, "<![CDATA[test reasoning]]>")
# Strip whitespace before comparing
self.assertEqual(reasoning_elem.text.strip(), "<![CDATA[test reasoning]]>")
def test_multiple_entries(self):
"""Test handling multiple entries in working memory."""
@@ -186,4 +186,4 @@ class BaseAgentTest(unittest.TestCase):
self.assertEqual(len(reasoning_elems), 3)
for i, elem in enumerate(reasoning_elems):
self.assertEqual(elem.get("id"), f"id-{i}")
self.assertEqual(elem.text, f"<![CDATA[test {i}]]>")
self.assertEqual(elem.text.strip(), f"<![CDATA[test {i}]]>")