From 9138bafdf58f537b608e18082d31b76596d2b18f Mon Sep 17 00:00:00 2001 From: Elia Date: Thu, 5 Mar 2026 01:17:27 +0100 Subject: [PATCH] fix: move import to top-level in bigquery-data-agent test --- python/agents/bigquery-data-agent/tests/test_agent.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/python/agents/bigquery-data-agent/tests/test_agent.py b/python/agents/bigquery-data-agent/tests/test_agent.py index 85391c0df..bafc38b6b 100644 --- a/python/agents/bigquery-data-agent/tests/test_agent.py +++ b/python/agents/bigquery-data-agent/tests/test_agent.py @@ -1,6 +1,10 @@ -import pytest +from google.adk.plugins.bigquery_agent_analytics_plugin import ( + BigQueryAgentAnalyticsPlugin, +) + from bigquery_data_agent.agent import app + def test_app_initialization(): """Test that the application and agent initialize correctly.""" assert app is not None @@ -18,8 +22,6 @@ def test_plugins_and_tools(): def test_plugin_configuration(): """Test that the BigQuery analytics plugin is properly configured.""" - from google.adk.plugins.bigquery_agent_analytics_plugin import BigQueryAgentAnalyticsPlugin - # Assert that at least one plugin is the analytics plugin analytics_plugins = [p for p in app.plugins if isinstance(p, BigQueryAgentAnalyticsPlugin)] assert len(analytics_plugins) == 1