diff --git a/dbgpt/datasource/rdbms/tests/test_conn_duckdb.py b/dbgpt/datasource/rdbms/tests/test_conn_duckdb.py index 356e2bff9..5eabbb074 100644 --- a/dbgpt/datasource/rdbms/tests/test_conn_duckdb.py +++ b/dbgpt/datasource/rdbms/tests/test_conn_duckdb.py @@ -17,25 +17,25 @@ def 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_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"] +def test_get_users(db): + assert db.get_users() == [] + + +def test_get_table_names(db): + assert list(db.get_table_names()) == [] + + +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"] diff --git a/setup.py b/setup.py index 915c53513..bb6a799e5 100644 --- a/setup.py +++ b/setup.py @@ -449,7 +449,7 @@ def core_requires(): # https://github.com/eosphoros-ai/DB-GPT/issues/551 # TODO: remove pandas dependency # alpine can't install pandas by default - "pandas>=2.2.2", + "pandas==2.0.3", # numpy should less than 2.0.0 "numpy>=1.21.0,<2.0.0", ] @@ -472,7 +472,7 @@ def core_requires(): # find a new toolkit. "pympler", "duckdb", - "duckdb-engine", + "duckdb-engine==0.9.1", # lightweight python library for scheduling jobs "schedule", # For datasource subpackage