def test_set_udaf_method_with_agg_class(self):
sqldf = SQLDF(locals())
conn = sqldf.conn
self.default_df.to_sql("df", conn)
sqldf._set_udaf(self.default_udafs)
self.assertEqual(list(conn.execute("select udaf1(label) from df;")),
[(3, )])
def test_set_udaf_method_with_agg_class(self):
sqldf = SQLDF(locals())
conn = sqldf.conn
self.default_df.to_sql("df", conn)
sqldf._set_udaf(self.default_udafs)
self.assertEqual(
list(
conn.execute("select udaf1(label) from df;")), [
(3,)])
def test_set_udaf_method_with_agg_function(self):
sqldf = SQLDF(locals())
conn = sqldf.conn
self.default_df.to_sql("df", conn)
def agg_func(values):
return len(values)
sqldf._set_udaf({"mycount": agg_func})
self.assertEqual(list(conn.execute("select mycount(label) from df;")),
[(3, )])
def test_set_udaf_method_with_agg_function(self):
sqldf = SQLDF(locals())
conn = sqldf.conn
self.default_df.to_sql("df", conn)
def agg_func(values):
return len(values)
sqldf._set_udaf({"mycount": agg_func})
self.assertEqual(
list(
conn.execute("select mycount(label) from df;")), [
(3,)])
