feat(postgres): implement array sort

ibis-project · Aug 5, 2023 · 4791cb4 · 4791cb4
1 parent df14997
commit 4791cb4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/ibis/backends/postgres/registry.py b/ibis/backends/postgres/registry.py
@@ -555,6 +555,11 @@ def _unnest(t, op):
     return sa.func.row(*derived.c)
 
 
+def _array_sort(arg):
+    flat = sa.func.unnest(arg).column_valued()
+    return sa.func.array(sa.select(flat).order_by(flat).scalar_subquery())
+
+
 operation_registry.update(
     {
         ops.Literal: _literal,
@@ -716,5 +721,6 @@ def _unnest(t, op):
             lambda arg: sa.extract("microsecond", arg) % 1_000_000, 1
         ),
         ops.Levenshtein: fixed_arity(sa.func.levenshtein, 2),
+        ops.ArraySort: fixed_arity(_array_sort, 1),
     }
 )
diff --git a/ibis/backends/tests/test_array.py b/ibis/backends/tests/test_array.py
@@ -656,7 +656,6 @@ def test_array_unique(backend, con):
         "mssql",
         "pandas",
         "polars",
-        "postgres",
         "snowflake",
     ],
     raises=com.OperationNotDefinedError,