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

Fixed stdin buffer length in context

Browse Source
This commit is contained in:
Niels Geens
2024-11-14 13:36:20 +01:00
parent 834ce12e8f
commit 325870a81a
6 changed files with 27 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/system_metrics_test.py
View File

@@ -96,7 +96,7 @@ class SystemMetricsTest(unittest.TestCase):
def test_initial_context(self):
"""Test initial context generation"""
context = self.metrics.generate_context(0.5)
context = self.metrics.generate_context(0.5, 10)
# Verify it's an XML element
self.assertIsInstance(context, ET.Element)
@@ -125,13 +125,13 @@ class SystemMetricsTest(unittest.TestCase):
])
# Validate stdin buffer size
self.assertEqual(context.get("stdin"), "0")
self.assertEqual(context.get("stdin"), 10)
def test_multiple_context_generations(self):
"""Test multiple context generations have different timestamps"""
context1 = self.metrics.generate_context(0.5)
context1 = self.metrics.generate_context(0.5, 10)
time.sleep(1)
context2 = self.metrics.generate_context(0.5)
context2 = self.metrics.generate_context(0.5, 10)
self.assertNotEqual(
context1.get("time"),