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

Added timeout and limit to script entries

Browse Source
This commit is contained in:
Niels Geens
2024-11-13 17:48:39 +01:00
parent 65478e0b17
commit 834ce12e8f
21 changed files with 262 additions and 141 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
test/reasoning_entry_test.py
View File

@@ -12,7 +12,7 @@ class ReasoningEntryTest(unittest.TestCase):
def test_initialization(self):
"""Test entry initialization"""
content = "test reasoning"
entry = ReasoningEntry(content, self.test_id, self.test_timestamp)
entry = ReasoningEntry(self.test_id, self.test_timestamp, content)
self.assertEqual(entry.content, content)
self.assertEqual(entry.id, self.test_id)
@@ -21,7 +21,7 @@ class ReasoningEntryTest(unittest.TestCase):
def test_update_does_nothing(self):
"""Test that update operation has no effect"""
content = "test reasoning"
entry = ReasoningEntry(content, self.test_id, self.test_timestamp)
entry = ReasoningEntry(self.test_id, self.test_timestamp, content)
# Store initial state
initial_content = entry.generate_context().text
@@ -35,7 +35,7 @@ class ReasoningEntryTest(unittest.TestCase):
def test_generate_context(self):
"""Test XML context generation"""
content = "test reasoning"
entry = ReasoningEntry(content, self.test_id, self.test_timestamp)
entry = ReasoningEntry(self.test_id, self.test_timestamp, content)
element = entry.generate_context()
@@ -47,14 +47,14 @@ class ReasoningEntryTest(unittest.TestCase):
def test_special_characters(self):
"""Test handling reasoning text with special characters"""
content = "Special reasoning: \n\t\r\'\"\\"
entry = ReasoningEntry(content, self.test_id, self.test_timestamp)
entry = ReasoningEntry(self.test_id, self.test_timestamp, content)
element = entry.generate_context()
self.assertEqual(element.text, f"{content}")
def test_empty_content(self):
"""Test handling empty reasoning text"""
entry = ReasoningEntry("", self.test_id, self.test_timestamp)
entry = ReasoningEntry(self.test_id, self.test_timestamp, "")
element = entry.generate_context()
self.assertEqual(element.text, "")
@@ -62,7 +62,7 @@ class ReasoningEntryTest(unittest.TestCase):
def test_large_content(self):
"""Test handling large reasoning text"""
content = "x" * 10000
entry = ReasoningEntry(content, self.test_id, self.test_timestamp)
entry = ReasoningEntry(self.test_id, self.test_timestamp, content)
element = entry.generate_context()
self.assertEqual(element.text, f"{content}")
@@ -72,7 +72,7 @@ class ReasoningEntryTest(unittest.TestCase):
content = """Line 1
Line 2
Line 3"""
entry = ReasoningEntry(content, self.test_id, self.test_timestamp)
entry = ReasoningEntry(self.test_id, self.test_timestamp, content)
element = entry.generate_context()
self.assertEqual(element.text, f"{content}")