test(flink): revise WIP marks in aggregation tests

ibis-project · Nov 6, 2023 · d87bd8f · d87bd8f
1 parent d36ed1c
commit d87bd8f
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/ibis/backends/tests/test_aggregation.py b/ibis/backends/tests/test_aggregation.py
@@ -67,6 +67,7 @@ def mean_udf(s):
                     "trino",
                     "druid",
                     "oracle",
+                    "flink",
                 ],
                 raises=com.OperationNotDefinedError,
             ),
@@ -75,7 +76,6 @@ def mean_udf(s):
                 reason="no udf support",
                 raises=com.OperationNotDefinedError,
             ),
-            pytest.mark.notimpl(["flink"], "WIP", raises=com.OperationNotDefinedError),
         ],
     ),
     param(lambda t: t.double_col.min(), lambda t: t.double_col.min(), id="min"),
@@ -98,10 +98,10 @@ def mean_udf(s):
                     "trino",
                     "druid",
                     "oracle",
+                    "flink",
                 ],
                 raises=com.OperationNotDefinedError,
             ),
-            pytest.mark.notimpl(["flink"], "WIP", raises=com.OperationNotDefinedError),
         ],
     ),
     param(
@@ -130,15 +130,13 @@ def mean_udf(s):
     "mssql",
     "druid",
     "oracle",
+    "flink",
 ]
 argidx_grouped_marks = ["dask"] + argidx_not_grouped_marks
 
 
 def make_argidx_params(marks):
-    marks = [
-        pytest.mark.notyet(marks, raises=com.OperationNotDefinedError),
-        pytest.mark.notimpl(["flink"], "WIP", raises=com.OperationNotDefinedError),
-    ]
+    marks = pytest.mark.notyet(marks, raises=com.OperationNotDefinedError)
     return [
         param(
             lambda t: t.timestamp_col.argmin(t.id),