Skip to content

Commit 9d20f3b

Browse files
committed
Update return types on unit tests
1 parent 6011e65 commit 9d20f3b

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

python/tests/test_functions.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -103,8 +103,11 @@ def test_lit_arith(df):
103103
result = df.collect()
104104
assert len(result) == 1
105105
result = result[0]
106+
106107
assert result.column(0) == pa.array([5, 6, 7])
107-
assert result.column(1) == pa.array(["Hello!", "World!", "!!"])
108+
assert result.column(1) == pa.array(
109+
["Hello!", "World!", "!!"], type=pa.string_view()
110+
)
108111

109112

110113
def test_math_functions():
@@ -661,7 +664,7 @@ def test_array_function_obj_tests(stmt, py_expr):
661664
),
662665
(
663666
f.concat(column("a").cast(pa.string()), literal("?")),
664-
pa.array(["Hello?", "World?", "!?"]),
667+
pa.array(["Hello?", "World?", "!?"], type=pa.string_view()),
665668
),
666669
(f.initcap(column("c")), pa.array(["Hello ", " World ", " !"])),
667670
(f.left(column("a"), literal(3)), pa.array(["Hel", "Wor", "!"])),
@@ -871,8 +874,8 @@ def test_temporal_functions(df):
871874
result = df.collect()
872875
assert len(result) == 1
873876
result = result[0]
874-
assert result.column(0) == pa.array([12, 6, 7], type=pa.float64())
875-
assert result.column(1) == pa.array([2022, 2027, 2020], type=pa.float64())
877+
assert result.column(0) == pa.array([12, 6, 7], type=pa.int32())
878+
assert result.column(1) == pa.array([2022, 2027, 2020], type=pa.int32())
876879
assert result.column(2) == pa.array(
877880
[datetime(2022, 12, 1), datetime(2027, 6, 1), datetime(2020, 7, 1)],
878881
type=pa.timestamp("us"),
@@ -904,7 +907,7 @@ def test_temporal_functions(df):
904907
assert result.column(9) == pa.array(
905908
[datetime(2023, 9, 7, 5, 6, 14, 523952)] * 3, type=pa.timestamp("us")
906909
)
907-
assert result.column(10) == pa.array([31, 26, 2], type=pa.float64())
910+
assert result.column(10) == pa.array([31, 26, 2], type=pa.int32())
908911

909912

910913
def test_case(df):

0 commit comments

Comments
 (0)