diff options
-rw-r--r-- | tests/test_smart_completion_multiple_schemata.py | 2 | ||||
-rw-r--r-- | tests/test_smart_completion_public_schema_only.py | 4 |
2 files changed, 4 insertions, 2 deletions
diff --git a/tests/test_smart_completion_multiple_schemata.py b/tests/test_smart_completion_multiple_schemata.py index 9125f124..9fff2a17 100644 --- a/tests/test_smart_completion_multiple_schemata.py +++ b/tests/test_smart_completion_multiple_schemata.py @@ -229,7 +229,7 @@ def test_suggested_column_names_in_function(completer): completer, "SELECT MAX( from custom.products", len("SELECT MAX(") ) assert completions_to_set(result) == completions_to_set( - testdata.columns("products", "custom") + testdata.columns_functions_and_keywords("products", "custom") ) diff --git a/tests/test_smart_completion_public_schema_only.py b/tests/test_smart_completion_public_schema_only.py index 63f1e80e..bb371543 100644 --- a/tests/test_smart_completion_public_schema_only.py +++ b/tests/test_smart_completion_public_schema_only.py @@ -287,7 +287,9 @@ def test_suggested_cased_always_qualified_column_names(completer): @parametrize("completer", completers(casing=False, qualify=no_qual)) def test_suggested_column_names_in_function(completer): result = get_result(completer, "SELECT MAX( from users", len("SELECT MAX(")) - assert completions_to_set(result) == completions_to_set(testdata.columns("users")) + assert completions_to_set(result) == completions_to_set( + (testdata.columns_functions_and_keywords("users")) + ) @parametrize("completer", completers(casing=False)) |