apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-24 04:31:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Fix tests

Browse source
This commit is contained in:
Cyber MacGeddon 2025-09-25 21:29:47 +01:00
parent 25b2617df2
commit 7e7b9beba7
1 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tests/integration/test_text_completion_integration.py
View file

@ -60,13 +60,13 @@ class TestTextCompletionIntegration:
"""Create text completion processor with test configuration""" """Create text completion processor with test configuration"""
# Create a minimal processor instance for testing generate_content # Create a minimal processor instance for testing generate_content
processor = MagicMock() processor = MagicMock()
processor.model = processor_config["model"] processor.default_model = processor_config["model"]
processor.temperature = processor_config["temperature"] processor.temperature = processor_config["temperature"]
processor.max_output = processor_config["max_output"] processor.max_output = processor_config["max_output"]
# Add the actual generate_content method from Processor class # Add the actual generate_content method from Processor class
processor.generate_content = Processor.generate_content.__get__(processor, Processor) processor.generate_content = Processor.generate_content.__get__(processor, Processor)
return processor return processor
@pytest.mark.asyncio @pytest.mark.asyncio
@ -112,11 +112,11 @@ class TestTextCompletionIntegration:
for config in test_configs: for config in test_configs:
# Arrange - Create minimal processor mock # Arrange - Create minimal processor mock
processor = MagicMock() processor = MagicMock()
processor.model = config['model'] processor.default_model = config['model']
processor.temperature = config['temperature'] processor.temperature = config['temperature']
processor.max_output = config['max_output'] processor.max_output = config['max_output']
processor.openai = mock_openai_client processor.openai = mock_openai_client
# Add the actual generate_content method # Add the actual generate_content method
processor.generate_content = Processor.generate_content.__get__(processor, Processor) processor.generate_content = Processor.generate_content.__get__(processor, Processor)
@ -242,7 +242,7 @@ class TestTextCompletionIntegration:
processors = [] processors = []
for i in range(5): for i in range(5):
processor = MagicMock() processor = MagicMock()
processor.model = processor_config["model"] processor.default_model = processor_config["model"]
processor.temperature = processor_config["temperature"] processor.temperature = processor_config["temperature"]
processor.max_output = processor_config["max_output"] processor.max_output = processor_config["max_output"]
processor.openai = mock_openai_client processor.openai = mock_openai_client
@ -348,7 +348,7 @@ class TestTextCompletionIntegration:
"""Test that model parameters are correctly passed to OpenAI API""" """Test that model parameters are correctly passed to OpenAI API"""
# Arrange # Arrange
processor = MagicMock() processor = MagicMock()
processor.model = "gpt-4" processor.default_model = "gpt-4"
processor.temperature = 0.8 processor.temperature = 0.8
processor.max_output = 2048 processor.max_output = 2048
processor.openai = mock_openai_client processor.openai = mock_openai_client