Skip to content

Commit dcaf0d6

Browse files
committed
Revert "fix: update test_arrow_cast to cast literals to string type for arrow_cast function"
This reverts commit 856ff8c.
1 parent 856ff8c commit dcaf0d6

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

python/tests/test_functions.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323

2424
from datafusion import SessionContext, column
2525
from datafusion import functions as f
26-
from datafusion import literal, expr
26+
from datafusion import literal
2727

2828
np.seterr(invalid="ignore")
2929

@@ -907,10 +907,8 @@ def test_temporal_functions(df):
907907

908908
def test_arrow_cast(df):
909909
df = df.select(
910-
f.arrow_cast(column("a"), literal("Float64").cast(pa.string())).alias(
911-
"a_as_float"
912-
),
913-
f.arrow_cast(column("a"), literal("Int32").cast(pa.string())).alias("a_as_int"),
910+
f.arrow_cast(column("a"), literal("Float64")).alias("a_as_float"),
911+
f.arrow_cast(column("a"), literal("Int32")).alias("a_as_int"),
914912
)
915913
result = df.collect()
916914
assert len(result) == 1

0 commit comments

Comments
 (0)