Skip to content

Commit

Permalink
fix python test
Browse files Browse the repository at this point in the history
  • Loading branch information
goldmedal committed Nov 27, 2024
1 parent a7df53f commit 2db41a8
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions wren-core-py/tests/test_modeling_core.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,14 +46,14 @@ def test_read_function_list():
path = "tests/functions.csv"
session_context = SessionContext(manifest_str, path)
functions = session_context.get_available_functions()
assert len(functions) == 271
assert len(functions) == 272

rewritten_sql = session_context.transform_sql("SELECT add_two(c_custkey) FROM my_catalog.my_schema.customer")
assert rewritten_sql == 'SELECT add_two(customer.c_custkey) FROM (SELECT customer.c_custkey FROM (SELECT customer.c_custkey AS c_custkey FROM main.customer) AS customer) AS customer'

session_context = SessionContext(manifest_str, None)
functions = session_context.get_available_functions()
assert len(functions) == 269
assert len(functions) == 270


def test_get_available_functions():
Expand Down

0 comments on commit 2db41a8

Please sign in to comment.