|
|
""" |
|
|
Basic Tests for GAIA Agent - Stage 1 Validation |
|
|
Author: @mangubee |
|
|
Date: 2026-01-01 |
|
|
|
|
|
Tests for Stage 1: Foundation Setup |
|
|
- Agent initialization |
|
|
- StateGraph compilation |
|
|
- Basic question processing |
|
|
""" |
|
|
|
|
|
import pytest |
|
|
from src.agent import GAIAAgent |
|
|
from src.config import Settings |
|
|
|
|
|
|
|
|
class TestAgentInitialization: |
|
|
"""Test agent initialization and configuration.""" |
|
|
|
|
|
def test_agent_init(self): |
|
|
"""Test that agent can be initialized without errors.""" |
|
|
agent = GAIAAgent() |
|
|
assert agent is not None |
|
|
assert agent.graph is not None |
|
|
print("✓ Agent initialization successful") |
|
|
|
|
|
def test_settings_load(self): |
|
|
"""Test that settings can be loaded.""" |
|
|
settings = Settings() |
|
|
assert settings is not None |
|
|
assert settings.max_retries == 3 |
|
|
assert settings.question_timeout == 1020 |
|
|
print("✓ Settings loaded successfully") |
|
|
|
|
|
|
|
|
class TestBasicExecution: |
|
|
"""Test basic agent execution with placeholder logic.""" |
|
|
|
|
|
def test_simple_question(self): |
|
|
"""Test agent with simple question.""" |
|
|
agent = GAIAAgent() |
|
|
answer = agent("What is 2+2?") |
|
|
assert isinstance(answer, str) |
|
|
assert len(answer) > 0 |
|
|
print(f"✓ Agent returned answer: {answer}") |
|
|
|
|
|
def test_long_question(self): |
|
|
"""Test agent with longer question.""" |
|
|
agent = GAIAAgent() |
|
|
long_question = "Explain the significance of the French Revolution in European history and its impact on modern democracy." |
|
|
answer = agent(long_question) |
|
|
assert isinstance(answer, str) |
|
|
assert len(answer) > 0 |
|
|
print(f"✓ Agent handled long question, returned: {answer[:50]}...") |
|
|
|
|
|
def test_multiple_calls(self): |
|
|
"""Test that agent can handle multiple sequential calls.""" |
|
|
agent = GAIAAgent() |
|
|
questions = [ |
|
|
"What is the capital of France?", |
|
|
"Who wrote Romeo and Juliet?", |
|
|
"What is 10 * 5?" |
|
|
] |
|
|
for q in questions: |
|
|
answer = agent(q) |
|
|
assert isinstance(answer, str) |
|
|
assert len(answer) > 0 |
|
|
print(f"✓ Agent successfully processed {len(questions)} questions") |
|
|
|
|
|
|
|
|
class TestStateGraphStructure: |
|
|
"""Test StateGraph structure and nodes.""" |
|
|
|
|
|
def test_graph_has_nodes(self): |
|
|
"""Test that compiled graph has expected nodes.""" |
|
|
agent = GAIAAgent() |
|
|
|
|
|
|
|
|
assert agent.graph is not None |
|
|
print("✓ StateGraph compiled with expected structure") |
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
print("\n" + "="*70) |
|
|
print("GAIA Agent - Stage 1 Basic Tests") |
|
|
print("="*70 + "\n") |
|
|
|
|
|
|
|
|
test_init = TestAgentInitialization() |
|
|
test_init.test_agent_init() |
|
|
test_init.test_settings_load() |
|
|
|
|
|
test_exec = TestBasicExecution() |
|
|
test_exec.test_simple_question() |
|
|
test_exec.test_long_question() |
|
|
test_exec.test_multiple_calls() |
|
|
|
|
|
test_graph = TestStateGraphStructure() |
|
|
test_graph.test_graph_has_nodes() |
|
|
|
|
|
print("\n" + "="*70) |
|
|
print("✓ All Stage 1 tests passed!") |
|
|
print("="*70 + "\n") |
|
|
|
