def list_udfs(self, database=None, like=None):
"""
Lists all UDFs associated with given database
Parameters
----------
database : string
like : string for searching (optional)
"""
if not database:
database = self.current_database
statement = ddl.ListFunction(database, like=like, aggregate=False)
with self._execute(statement, results=True) as cur:
result = self._get_udfs(cur)
return result
def test_list_udafs_like(self):
stmt = ddl.ListFunction('test', like='identity', aggregate=True)
result = stmt.compile()
expected = "SHOW AGGREGATE FUNCTIONS IN test LIKE 'identity'"
assert result == expected
def test_list_udafs(self):
stmt = ddl.ListFunction('test', aggregate=True)
result = stmt.compile()
expected = 'SHOW AGGREGATE FUNCTIONS IN test'
assert result == expected
def test_list_udfs_like(self):
stmt = ddl.ListFunction('test', like='identity')
result = stmt.compile()
expected = "SHOW FUNCTIONS IN test LIKE 'identity'"
assert result == expected
def test_list_udf(self):
stmt = ddl.ListFunction('test')
result = stmt.compile()
expected = 'SHOW FUNCTIONS IN test'
assert result == expected
