fix: update test_arrow_cast to use literal casting for data types

kosiew · kosiew · commit 11071e684692 · 2024-11-26T17:41:15.000+08:00
diff --git a/python/tests/test_functions.py b/python/tests/test_functions.py
@@ -907,8 +907,10 @@ 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").cast(pa.string())).alias(
+            "a_as_float"
+        ),
+        f.arrow_cast(column("a"), literal("Int32").cast(pa.string())).alias("a_as_int"),
     )
     result = df.collect()
     assert len(result) == 1
