add dask diff method (#793)

narwhals-dev · Aug 15, 2024 · 166c49e · 166c49e
1 parent cc01cae
commit 166c49e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/narwhals/_dask/expr.py b/narwhals/_dask/expr.py
@@ -485,6 +485,13 @@ def clip(
             returns_scalar=False,
         )
 
+    def diff(self: Self) -> Self:
+        return self._from_call(
+            lambda _input: _input.diff(),
+            "diff",
+            returns_scalar=False,
+        )
+
     def n_unique(self: Self) -> Self:
         return self._from_call(
             lambda _input: _input.nunique(dropna=False),

diff --git a/tests/expr_and_series/diff_test.py b/tests/expr_and_series/diff_test.py
@@ -15,8 +15,6 @@
 
 
 def test_diff(constructor: Any, request: Any) -> None:
-    if "dask" in str(constructor):
-        request.applymarker(pytest.mark.xfail)
     if "pyarrow_table_constructor" in str(constructor) and parse_version(
         pa.__version__
     ) < (13,):