docs: add dbgpt_hub usage documents (#955)

This commit is contained in:
magic.chen
2023-12-20 10:18:22 +08:00
committed by GitHub
parent ba8fa8774d
commit aec124a5f1
13 changed files with 339 additions and 67 deletions

View File

@@ -0,0 +1,44 @@
"""
Run unit test with command: pytest dbgpt/datasource/rdbms/tests/test_conn_duckdb.py
"""
import pytest
import tempfile
from dbgpt.datasource.rdbms.conn_duckdb import DuckDbConnect
@pytest.fixture
def db():
temp_db_file = tempfile.NamedTemporaryFile(delete=False)
temp_db_file.close()
conn = DuckDbConnect.from_file_path(temp_db_file.name + "duckdb.db")
yield conn
def test_get_users(db):
assert db.get_users() == []
def test_get_table_names(db):
assert list(db.get_table_names()) == []
def test_get_users(db):
assert db.get_users() == []
def test_get_charset(db):
assert db.get_charset() == "UTF-8"
def test_get_table_comments(db):
assert db.get_table_comments("test") == []
def test_table_simple_info(db):
assert db.table_simple_info() == []
def test_execute(db):
assert list(db.run("SELECT 42")[0]) == ["42"]