From f7998b596be6637a7a37e8cc813884df8b30c68e Mon Sep 17 00:00:00 2001 From: Jax Liu Date: Fri, 27 Dec 2024 16:57:07 +0800 Subject: [PATCH] upgrade function number --- ibis-server/tests/conftest.py | 2 +- .../tests/routers/v3/connector/postgres/test_functions.py | 2 +- wren-core-py/tests/test_modeling_core.py | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/ibis-server/tests/conftest.py b/ibis-server/tests/conftest.py index 28bfee41f..bc60a822d 100644 --- a/ibis-server/tests/conftest.py +++ b/ibis-server/tests/conftest.py @@ -11,7 +11,7 @@ def file_path(path: str) -> str: return os.path.join(os.path.dirname(__file__), path) -DATAFUSION_FUNCTION_COUNT = 270 +DATAFUSION_FUNCTION_COUNT = 271 @pytest.fixture(scope="session") diff --git a/ibis-server/tests/routers/v3/connector/postgres/test_functions.py b/ibis-server/tests/routers/v3/connector/postgres/test_functions.py index 5f8beca82..5cca621b7 100644 --- a/ibis-server/tests/routers/v3/connector/postgres/test_functions.py +++ b/ibis-server/tests/routers/v3/connector/postgres/test_functions.py @@ -57,7 +57,7 @@ async def test_function_list(client): response = await client.get(url=f"{base_url}/functions") assert response.status_code == 200 result = response.json() - assert len(result) == DATAFUSION_FUNCTION_COUNT + 35 + assert len(result) == DATAFUSION_FUNCTION_COUNT + 34 the_func = next(filter(lambda x: x["name"] == "extract", result)) assert the_func == { "name": "extract", diff --git a/wren-core-py/tests/test_modeling_core.py b/wren-core-py/tests/test_modeling_core.py index f94b32510..a6563082e 100644 --- a/wren-core-py/tests/test_modeling_core.py +++ b/wren-core-py/tests/test_modeling_core.py @@ -106,7 +106,7 @@ def test_read_function_list(): path = "tests/functions.csv" session_context = SessionContext(manifest_str, path) functions = session_context.get_available_functions() - assert len(functions) == 272 + assert len(functions) == 273 rewritten_sql = session_context.transform_sql( "SELECT add_two(c_custkey) FROM my_catalog.my_schema.customer" @@ -118,7 +118,7 @@ def test_read_function_list(): session_context = SessionContext(manifest_str, None) functions = session_context.get_available_functions() - assert len(functions) == 270 + assert len(functions) == 271 def test_get_available_functions():