Add list_repeat

Author: kosiew

python/datafusion/functions.py

@@ -164,6 +164,7 @@
     "list_prepend",
     "list_push_back",
     "list_push_front",
+    "list_repeat",
     "list_remove",
     "list_remove_all",
     "list_remove_n",
@@ -1384,6 +1385,14 @@ def array_repeat(element: Expr, count: Expr) -> Expr:
     return Expr(f.array_repeat(element.expr, count.expr))
 
 
+def list_repeat(element: Expr, count: Expr) -> Expr:
+    """Returns an array containing ``element`` ``count`` times.
+
+    This is an alias for :py:func:`array_repeat`.
+    """
+    return array_repeat(element, count)
+
+
 def array_replace(array: Expr, from_val: Expr, to_val: Expr) -> Expr:
     """Replaces the first occurrence of ``from_val`` with ``to_val``."""
     return Expr(f.array_replace(array.expr, from_val.expr, to_val.expr))

python/tests/test_functions.py

@@ -445,6 +445,10 @@ def py_flatten(arr):
             lambda col: f.array_repeat(col, literal(2)),
             lambda data: [[arr] * 2 for arr in data],
         ],
+        [
+            lambda col: f.list_repeat(col, literal(2)),
+            lambda data: [[arr] * 2 for arr in data],
+        ],
         [
             lambda col: f.array_replace(col, literal(3.0), literal(4.0)),
             lambda data: [py_arr_replace(arr, 3.0, 4.0, 1) for arr in data],