Fix test

icexelloss · Jul 17, 2023 · 6473fed · 6473fed
1 parent 408dc7e
commit 6473fed
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/python/pyarrow/_compute.pyx b/python/pyarrow/_compute.pyx
@@ -2861,9 +2861,9 @@ def register_vector_function(func, function_name, function_doc, in_types, out_ty
     >>> answer
     <pyarrow.lib.DoubleArray object at ...>
     [
-        0.3333333333333333,
-        0.6666666666666666,
-        1
+      0.3333333333333333,
+      0.6666666666666666,
+      1
     ]
     """
     return _register_user_defined_function(get_register_vector_function(),

diff --git a/python/pyarrow/tests/test_udf.py b/python/pyarrow/tests/test_udf.py
@@ -841,15 +841,15 @@ def test_hash_agg_random(sum_agg_func_fixture):
 def test_vector_basic(unary_vector_func_fixture):
     arr = pa.array([10.0, 20.0, 30.0, 40.0, 50.0], pa.float64())
     result = pc.call_function("y=pct_rank(x)", [arr])
-    expected = unary_vector_func_fixture(None, arr)
+    expected = unary_vector_func_fixture[0](None, arr)
     assert result == expected
 
 
 @pytest.mark.pandas
 def test_vector_empty(unary_vector_func_fixture):
     arr = pa.array([1], pa.float64())
     result = pc.call_function("y=pct_rank(x)", [arr])
-    expected = unary_vector_func_fixture(None, arr)
+    expected = unary_vector_func_fixture[0](None, arr)
     assert result == expected