fix: update test_arrow_cast to use literal for data type parameters

kosiew · kosiew · commit 8c7e2f8ac6c6 · 2024-11-26T18:06:29.000+08:00
diff --git a/python/tests/test_functions.py b/python/tests/test_functions.py
@@ -907,8 +907,8 @@ def test_temporal_functions(df):
 
 def test_arrow_cast(df):
     df = df.select(
-        f.arrow_cast(column("a"), "Float64").alias("a_as_float"),
-        f.arrow_cast(column("a"), "Int32").alias("a_as_int"),
+        f.arrow_cast(column("a"), literal("Float64")).alias("a_as_float"),
+        f.arrow_cast(column("a"), literal("Int32")).alias("a_as_int"),
     )
     result = df.collect()
     assert len(result) == 1
