zach
/
RA.Aid
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix expert unit tests.

This commit is contained in:
user 2024-12-23 12:38:38 -05:00
parent 04d45425ee
commit 9a283f3f43
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tests/ra_aid/tools/test_expert.py
View File

@ -86,19 +86,20 @@ def test_read_files_with_limit_encoding_error(temp_test_files):
def test_expert_context_management(): def test_expert_context_management():
"""Test expert context global state management.""" """Test expert context global state management."""
# Clear any existing context # Clear any existing context
expert_context.clear() expert_context['text'].clear()
expert_context['files'].clear()
# Test adding context # Test adding context
result1 = emit_expert_context("Test context 1") result1 = emit_expert_context("Test context 1")
assert "Context added" in result1 assert "Context added" in result1
assert len(expert_context) == 1 assert len(expert_context['text']) == 1
assert expert_context[0] == "Test context 1" assert expert_context['text'][0] == "Test context 1"
# Test adding multiple contexts # Test adding multiple contexts
result2 = emit_expert_context("Test context 2") result2 = emit_expert_context("Test context 2")
assert "Context added" in result2 assert "Context added" in result2
assert len(expert_context) == 2 assert len(expert_context['text']) == 2
assert expert_context[1] == "Test context 2" assert expert_context['text'][1] == "Test context 2"
# Test context accumulation # Test context accumulation
assert all(ctx in expert_context for ctx in ["Test context 1", "Test context 2"]) assert all(ctx in expert_context['text'] for ctx in ["Test context 1", "Test context 2"])