feat(datafusion): implement Limit

Author: cpcloud

ibis/backends/datafusion/compiler.py

@@ -107,6 +107,13 @@ def selection(op):
     return plan
 
 
+@translate.register(ops.Limit)
+def limit(op):
+    if op.offset:
+        raise NotImplementedError("DataFusion does not support offset")
+    return translate(op.table).limit(op.n)
+
+
 @translate.register(ops.Aggregation)
 def aggregation(op):
     table = translate(op.table)

ibis/backends/tests/test_api.py

@@ -2,7 +2,6 @@
 from pytest import param
 
 import ibis
-import ibis.common.exceptions as exc
 
 
 def test_backend_name(backend):
@@ -73,7 +72,6 @@ def test_tables_accessor_tab_completion(con):
     assert 'functional_alltypes' in keys
 
 
-@pytest.mark.notimpl(["datafusion"], raises=exc.OperationNotDefinedError)
 @pytest.mark.parametrize(
     "expr_fn",
     [