Skip to content

Commit 98e5f80

Browse files
committed
fix: update test_arrow_cast to use column 'b' and add debug prints for actual values
1 parent ba53bd1 commit 98e5f80

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

python/tests/test_functions.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -909,12 +909,17 @@ def test_temporal_functions(df):
909909

910910
def test_arrow_cast(df):
911911
df = df.select(
912-
f.arrow_cast(column("a"), utf8_literal("Float64")).alias("a_as_float"),
913-
f.arrow_cast(column("a"), utf8_literal("Int32")).alias("a_as_int"),
912+
f.arrow_cast(column("b"), utf8_literal("Float64")).alias("a_as_float"),
913+
f.arrow_cast(column("b"), utf8_literal("Int32")).alias("a_as_int"),
914914
)
915915
result = df.collect()
916916
assert len(result) == 1
917917
result = result[0]
918+
919+
# Print the actual values in the columns for debugging
920+
print("Actual values in a_as_float:", result.column(0).to_pylist())
921+
print("Actual values in a_as_int:", result.column(1).to_pylist())
922+
918923
assert result.column(0) == pa.array([1.0, 2.0, 3.0], type=pa.float64())
919924
assert result.column(1) == pa.array([1, 2, 3], type=pa.int32())
920925

0 commit comments

Comments
 (0)