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,)])