fix tests

Author: dkundel-openai

examples/financial_research_agent/manager.py

@@ -42,16 +42,14 @@ async def run(self, query: str) -> None:
                 is_done=True,
                 hide_checkmark=True,
             )
-            self.printer.update_item(
-                "start", "Starting financial research...", is_done=True)
+            self.printer.update_item("start", "Starting financial research...", is_done=True)
             search_plan = await self._plan_searches(query)
             search_results = await self._perform_searches(search_plan)
             report = await self._write_report(query, search_results)
             verification = await self._verify_report(report)
 
             final_report = f"Report summary\n\n{report.short_summary}"
-            self.printer.update_item(
-                "final_report", final_report, is_done=True)
+            self.printer.update_item("final_report", final_report, is_done=True)
 
             self.printer.end()
 
@@ -76,8 +74,7 @@ async def _plan_searches(self, query: str) -> FinancialSearchPlan:
     async def _perform_searches(self, search_plan: FinancialSearchPlan) -> Sequence[str]:
         with custom_span("Search the web"):
             self.printer.update_item("searching", "Searching...")
-            tasks = [asyncio.create_task(self._search(item))
-                     for item in search_plan.searches]
+            tasks = [asyncio.create_task(self._search(item)) for item in search_plan.searches]
             results: list[str] = []
             num_completed = 0
             for task in asyncio.as_completed(tasks):
@@ -112,8 +109,7 @@ async def _write_report(self, query: str, search_results: Sequence[str]) -> Fina
             tool_description="Use to get a short write‑up of potential red flags",
             custom_output_extractor=_summary_extractor,
         )
-        writer_with_tools = writer_agent.clone(
-            tools=[fundamentals_tool, risk_tool])
+        writer_with_tools = writer_agent.clone(tools=[fundamentals_tool, risk_tool])
         self.printer.update_item("writing", "Thinking about report...")
         input_data = f"Original query: {query}\nSummarized search results: {search_results}"
         result = Runner.run_streamed(writer_with_tools, input_data)
@@ -126,8 +122,7 @@ async def _write_report(self, query: str, search_results: Sequence[str]) -> Fina
         next_message = 0
         async for _ in result.stream_events():
             if time.time() - last_update > 5 and next_message < len(update_messages):
-                self.printer.update_item(
-                    "writing", update_messages[next_message])
+                self.printer.update_item("writing", update_messages[next_message])
                 next_message += 1
                 last_update = time.time()
         self.printer.mark_item_done("writing")

examples/financial_research_agent/printer.py

@@ -10,6 +10,7 @@ class Printer:
     Simple wrapper to stream status updates. Used by the financial bot
     manager as it orchestrates planning, search and writing.
     """
+
     def __init__(self, console: Console) -> None:
         self.live = Live(console=console)
         self.items: dict[str, tuple[str, bool]] = {}

tests/voice/conftest.py

@@ -0,0 +1,14 @@
+import os
+import sys
+
+import pytest
+
+
+def pytest_collection_modifyitems(config, items):
+    if sys.version_info[:2] == (3, 9):
+        this_dir = os.path.dirname(__file__)
+        skip_marker = pytest.mark.skip(reason="Skipped on Python 3.9")
+
+        for item in items:
+            if item.fspath.dirname.startswith(this_dir):
+                item.add_marker(skip_marker)