Method test_with_function

tests/_ast/test_sql_visitor.py:525–530

(self)

        }

    def test_with_function(self) -> None:
        sql = """
        SELECT *, embedding(text) as text_embedding
        FROM prompts;
        """
        assert find_sql_refs(sql) == {SQLRef(table="prompts")}

    def test_with_schema(self) -> None:
        sql = "SELECT * FROM schema1.table1;"