Exemple #1
0
 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", )])
Exemple #2
0
 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",)])