From 38bad52414f1dee11337dcc6ffbe0d80273a54b6 Mon Sep 17 00:00:00 2001 From: Dominik Jain Date: Fri, 5 Dec 2025 20:46:44 +0100 Subject: [PATCH] Use context manager for agent fixture --- test/serena/test_serena_agent.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/test/serena/test_serena_agent.py b/test/serena/test_serena_agent.py index 30d4cb4b..bebab035 100644 --- a/test/serena/test_serena_agent.py +++ b/test/serena/test_serena_agent.py @@ -2,6 +2,7 @@ import json import logging import os import time +from collections.abc import Iterator import pytest @@ -54,14 +55,17 @@ def serena_config(): @pytest.fixture -def serena_agent(request: pytest.FixtureRequest, serena_config): +def serena_agent(request: pytest.FixtureRequest, serena_config) -> Iterator[SerenaAgent]: language = Language(request.param) if not language_tests_enabled(language): pytest.skip(f"Tests for language {language} are not enabled.") project_name = f"test_repo_{language}" - return SerenaAgent(project=project_name, serena_config=serena_config) + agent = SerenaAgent(project=project_name, serena_config=serena_config) + yield agent + # explicitly delete to free resources + del agent class TestSerenaAgent: