test(postgres): fix tests

ibis-project · Jun 30, 2024 · b381858 · b381858
1 parent 36005ea
commit b381858
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/ibis/backends/postgres/tests/test_functions.py b/ibis/backends/postgres/tests/test_functions.py
@@ -384,8 +384,7 @@ def test_coalesce_all_na_double(con):
 def test_numeric_builtins_work(alltypes, df):
     expr = alltypes.double_col.fill_null(0)
     result = expr.execute()
-    expected = df.double_col.fillna(0)
-    expected.name = "Coalesce()"
+    expected = df.double_col.fillna(0).rename(expr.get_name())
     tm.assert_series_equal(result, expected)
 
 

diff --git a/ibis/backends/risingwave/tests/test_functions.py b/ibis/backends/risingwave/tests/test_functions.py
@@ -215,8 +215,7 @@ def test_coalesce_all_na_double(con):
 def test_numeric_builtins_work(alltypes, df):
     expr = alltypes.double_col.fill_null(0)
     result = expr.execute()
-    expected = df.double_col.fillna(0)
-    expected.name = "Coalesce()"
+    expected = df.double_col.fillna(0).rename(expr.get_name())
     tm.assert_series_equal(result, expected)