Fix pretty printing
This commit is contained in:
@@ -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}]]>")
|
||||
|
||||
Reference in New Issue
Block a user