From 2e06cf52771f987150ce0776d07ee4cc2465b58e Mon Sep 17 00:00:00 2001 From: Yiorgis Gozadinos Date: Tue, 9 Dec 2025 18:33:13 +0200 Subject: [PATCH] Delete unecessary tests --- tests/graph/agui/test_server.py | 23 ---------------------- tests/graph/test_deep_qa.py | 31 ------------------------------ tests/graph/test_research_graph.py | 15 --------------- 3 files changed, 69 deletions(-) diff --git a/tests/graph/agui/test_server.py b/tests/graph/agui/test_server.py index 8a6c78be..24ca0d06 100644 --- a/tests/graph/agui/test_server.py +++ b/tests/graph/agui/test_server.py @@ -45,17 +45,6 @@ def test_run_agent_input_parsing(): assert input_data.state == {"question": "What is AI?"} -def test_run_agent_input_defaults(): - """Test RunAgentInput with defaults.""" - input_data = RunAgentInput() # type: ignore[call-arg] - - assert input_data.thread_id is None - assert input_data.run_id is None - assert input_data.state == {} - assert input_data.messages == [] - assert input_data.config == {} - - def test_format_sse_event(): """Test SSE event formatting.""" event = {"type": "TEST_EVENT", "data": "test"} @@ -236,15 +225,3 @@ def test_server_cors_headers(): assert ( "access-control-allow-origin" in response.headers or response.status_code == 200 ) - - -def test_agui_config_defaults(): - """Test AGUIConfig default values.""" - config = AGUIConfig() - - assert config.host == "0.0.0.0" - assert config.port == 8000 - assert config.cors_origins == ["*"] - assert config.cors_credentials is True - assert "GET" in config.cors_methods - assert "POST" in config.cors_methods diff --git a/tests/graph/test_deep_qa.py b/tests/graph/test_deep_qa.py index 69c9c078..a87629dc 100644 --- a/tests/graph/test_deep_qa.py +++ b/tests/graph/test_deep_qa.py @@ -2,7 +2,6 @@ import pytest from pydantic_ai.models.test import TestModel from haiku.rag.client import HaikuRAG -from haiku.rag.graph.common.models import SearchAnswer from haiku.rag.graph.deep_qa.dependencies import DeepQAContext from haiku.rag.graph.deep_qa.graph import build_deep_qa_graph from haiku.rag.graph.deep_qa.state import DeepQADeps, DeepQAState @@ -40,33 +39,3 @@ async def test_deep_qa_graph_end_to_end(monkeypatch, temp_db_path): assert isinstance(result.answer, str) client.close() - - -@pytest.mark.asyncio -async def test_deep_qa_context_operations(): - context = DeepQAContext(original_question="Test question?") - - assert context.original_question == "Test question?" - assert context.sub_questions == [] - assert context.qa_responses == [] - - context.sub_questions = ["Sub Q1", "Sub Q2"] - assert len(context.sub_questions) == 2 - - qa = SearchAnswer( - query="Sub Q1", - answer="Answer 1", - cited_chunks=["chunk_1", "chunk_2"], - confidence=0.9, - ) - context.add_qa_response(qa) - assert len(context.qa_responses) == 1 - assert context.qa_responses[0].query == "Sub Q1" - - -def test_deep_qa_state_initialization(): - context = DeepQAContext(original_question="Test?") - state = DeepQAState(context=context, max_sub_questions=5) - - assert state.context.original_question == "Test?" - assert state.max_sub_questions == 5 diff --git a/tests/graph/test_research_graph.py b/tests/graph/test_research_graph.py index fcca2967..badb79e6 100644 --- a/tests/graph/test_research_graph.py +++ b/tests/graph/test_research_graph.py @@ -8,21 +8,6 @@ from haiku.rag.graph.research.graph import build_research_graph from haiku.rag.graph.research.state import ResearchDeps, ResearchState -def test_build_graph_and_state(): - graph = build_research_graph() - assert graph is not None - - state = ResearchState( - context=ResearchContext( - original_question="What are the key features of haiku.rag?" - ), - max_iterations=1, - confidence_threshold=0.8, - ) - assert state.iterations == 0 - assert state.context.sub_questions == [] - - @pytest.mark.asyncio async def test_graph_end_to_end_with_test_model(monkeypatch, temp_db_path): """Test research graph with mocked LLM using AG-UI events."""