refactor fixtures

Author: AbhiPrasad

py/src/braintrust/wrappers/agno/_test_agno_helpers.py

@@ -5,21 +5,11 @@
 
 from inspect import isawaitable
 
-import pytest
-from braintrust import logger
-from braintrust.test_helpers import init_test_logger
 from braintrust.wrappers.agno.agent import wrap_agent
 
 PROJECT_NAME = "test-agno-app"
 
 
-@pytest.fixture
-def memory_logger():
-    init_test_logger(PROJECT_NAME)
-    with logger._internal_with_memory_background_logger() as bgl:
-        yield bgl
-
-
 class FakeMetrics:
     def __init__(self):
         self.input_tokens = 1

py/src/braintrust/wrappers/agno/conftest.py

@@ -0,0 +1,21 @@
+from pathlib import Path
+
+import pytest
+from braintrust import logger
+from braintrust.test_helpers import init_test_logger
+
+from ._test_agno_helpers import PROJECT_NAME
+
+
+@pytest.fixture
+def memory_logger():
+    init_test_logger(PROJECT_NAME)
+    with logger._internal_with_memory_background_logger() as bgl:
+        yield bgl
+
+
+@pytest.fixture(scope="module")
+def vcr_config():
+    return {
+        "cassette_library_dir": str(Path(__file__).parent.parent / "cassettes"),
+    }

py/src/braintrust/wrappers/agno/test_agno.py

@@ -4,8 +4,6 @@
 # pyright: reportUnknownParameterType=false
 # pyright: reportUnknownVariableType=false
 # pyright: reportUnknownArgumentType=false
-from pathlib import Path
-
 import pytest
 from braintrust.logger import start_span
 from braintrust.wrappers.agno import agent as agno_agent_module
@@ -27,13 +25,6 @@
 )
 
 
-@pytest.fixture(scope="module")
-def vcr_config():
-    return {
-        "cassette_library_dir": str(Path(__file__).parent.parent / "cassettes"),
-    }
-
-
 @pytest.mark.vcr
 def test_agno_simple_agent_execution(memory_logger):
     agent_module = pytest.importorskip("agno.agent")

py/src/braintrust/wrappers/agno/test_workflow.py

@@ -4,8 +4,6 @@
 # pyright: reportUnknownParameterType=false
 # pyright: reportUnknownVariableType=false
 # pyright: reportUnknownArgumentType=false
-from pathlib import Path
-
 import pytest
 from braintrust.wrappers.agno import setup_agno
 from braintrust.wrappers.agno.workflow import wrap_workflow
@@ -21,13 +19,6 @@
 )
 
 
-@pytest.fixture(scope="module")
-def vcr_config():
-    return {
-        "cassette_library_dir": str(Path(__file__).parent.parent / "cassettes"),
-    }
-
-
 @pytest.mark.vcr
 def test_agno_workflow_with_agent(memory_logger):
     agent_module = pytest.importorskip("agno.agent")