fix: exception assertion

Author: mesejo

ibis/backends/sql/compilers/pyspark.py

@@ -243,10 +243,10 @@ def visit_CountDistinctStar(self, op, *, arg, where):
         ]
         return self.f.count(sge.Distinct(expressions=cols))
 
-    def visit_FirstValue(self, op, *, arg):
+    def visit_FirstValue(self, op, *, arg, include_null):
         return sge.IgnoreNulls(this=self.f.first(arg))
 
-    def visit_LastValue(self, op, *, arg):
+    def visit_LastValue(self, op, *, arg, include_null):
         return sge.IgnoreNulls(this=self.f.last(arg))
 
     def visit_First(self, op, *, arg, where, order_by, include_null):

ibis/backends/tests/test_window.py

@@ -1116,27 +1116,33 @@ def test_first_last(backend):
         "bigquery",
         "oracle",
         "snowflake",
-        "exasol",
         "databricks",
         "pyspark",
     ],
     raises=AssertionError,
 )
+@pytest.mark.notyet(
+    ["polars"],
+    raises=com.OperationNotDefinedError,
+)
+@pytest.mark.notyet(
+    ["flink"],
+    raises=NotImplementedError,
+)
 @pytest.mark.notyet(
     [
         "mysql",
         "sqlite",
         "postgres",
         "datafusion",
-        "polars",
         "druid",
         "athena",
         "impala",
-        "flink",
         "mssql",
         "trino",
+        "exasol",
     ],
-    raises=AssertionError,
+    raises=Exception,
 )
 def test_first_last_include_nulls(backend):
     t = ibis.memtable({"a": (2, 2, 1, 1), "b": (None, 3, 5, None), "c": list(range(4))})