mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-08 23:58:34 +00:00
43 lines
1.6 KiB
Python
43 lines
1.6 KiB
Python
from importlib.metadata import PackageNotFoundError, version
|
|
|
|
import pytest
|
|
|
|
|
|
def test_databricks_sdk_import():
|
|
"""Test that databricks-sdk can be imported without errors.
|
|
|
|
This test verifies that the databricks-sdk dependency is properly installed
|
|
when using the tools extra, which is required by the databricks_query_tool.
|
|
"""
|
|
try:
|
|
import databricks.sdk
|
|
assert True
|
|
except ImportError as e:
|
|
pytest.fail(f"Failed to import databricks.sdk: {e}")
|
|
|
|
|
|
def test_databricks_sdk_version():
|
|
"""Test that the installed databricks-sdk version meets requirements.
|
|
|
|
The databricks-sdk should be version 0.46.0 or higher, but less than 1.0.0
|
|
to maintain compatibility with the databricks_query_tool.
|
|
"""
|
|
try:
|
|
sdk_version = version("databricks-sdk")
|
|
assert sdk_version >= "0.46.0", f"Databricks SDK version {sdk_version} is too old (< 0.46.0)"
|
|
assert sdk_version < "1.0.0", f"Databricks SDK version {sdk_version} is too new (>= 1.0.0)"
|
|
except PackageNotFoundError:
|
|
pytest.fail("databricks-sdk package not found")
|
|
|
|
|
|
def test_databricks_core_functionality():
|
|
"""Test core functionality of the Databricks SDK.
|
|
|
|
This test verifies that the WorkspaceClient class from the Databricks SDK
|
|
has the expected attributes for SQL functionality.
|
|
"""
|
|
from databricks.sdk import WorkspaceClient
|
|
|
|
# Verify that the WorkspaceClient class has the expected attributes for SQL operations
|
|
assert hasattr(WorkspaceClient, 'statement_execution'), "SQL statement execution functionality not available in WorkspaceClient"
|