feat(exasol): implement cov/corr

Author: jcrist

ibis/backends/sql/compilers/exasol.py

@@ -44,7 +44,6 @@ class ExasolCompiler(SQLGlotCompiler):
         ops.ArrayUnion,
         ops.ArrayZip,
         ops.BitwiseNot,
-        ops.Covariance,
         ops.CumeDist,
         ops.DateAdd,
         ops.DateSub,
@@ -120,6 +119,20 @@ def visit_NonNullLiteral(self, op, *, value, dtype):
     def visit_Date(self, op, *, arg):
         return self.cast(arg, dt.date)
 
+    def visit_Correlation(self, op, *, left, right, how, where):
+        if how == "sample":
+            raise com.UnsupportedOperationError(
+                "Exasol only implements `pop` correlation coefficient"
+            )
+
+        if (left_type := op.left.dtype).is_boolean():
+            left = self.cast(left, dt.Int32(nullable=left_type.nullable))
+
+        if (right_type := op.right.dtype).is_boolean():
+            right = self.cast(right, dt.Int32(nullable=right_type.nullable))
+
+        return self.agg.corr(left, right, where=where)
+
     def visit_GroupConcat(self, op, *, arg, sep, where, order_by):
         if where is not None:
             arg = self.if_(where, arg, NULL)

ibis/backends/tests/test_aggregation.py

@@ -1013,7 +1013,7 @@ def test_quantile(
                     raises=com.OperationNotDefinedError,
                 ),
                 pytest.mark.notyet(
-                    ["postgres", "duckdb", "snowflake", "risingwave"],
+                    ["postgres", "duckdb", "snowflake", "risingwave", "exasol"],
                     raises=com.UnsupportedOperationError,
                     reason="backend only implements population correlation coefficient",
                 ),
@@ -1114,7 +1114,7 @@ def test_quantile(
         ),
     ],
 )
-@pytest.mark.notimpl(["mssql", "exasol"], raises=com.OperationNotDefinedError)
+@pytest.mark.notimpl(["mssql"], raises=com.OperationNotDefinedError)
 def test_corr_cov(
     con,
     batting,