def test_set_udf_method(self): sqldf = SQLDF(locals()) conn = sqldf.conn self.default_df.to_sql("df", conn) sqldf._set_udf(self.default_udfs) self.assertEqual(list(conn.execute("select udf1(label) from df;")), [("l1", ), ("l2", ), ("l3", )])
def test_set_udf_method(self): sqldf = SQLDF(locals()) conn = sqldf.conn self.default_df.to_sql("df", conn) sqldf._set_udf(self.default_udfs) self.assertEqual( list( conn.execute("select udf1(label) from df;")), [ ("l1",), ("l2",), ("l3",)])