@@ -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
110113def 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
910913def test_case (df ):
0 commit comments