chore: remove remaning use of .label in tests

ibis-project · Dec 18, 2024 · 0e1d154 · 0e1d154
1 parent 5057012
commit 0e1d154
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/ibis/backends/impala/tests/test_bucket_histogram.py b/ibis/backends/impala/tests/test_bucket_histogram.py
@@ -81,8 +81,8 @@ def test_bucket_assign_labels(table, snapshot):
     bucket = table.f.bucket(buckets, include_under=True)
 
     size = table.group_by(bucket.name("tier")).size()
-    labelled = size.tier.label(
-        ["Under 0", "0 to 10", "10 to 25", "25 to 50"], nulls="error"
+    labelled = size.tier.cases(
+        *enumerate(["Under 0", "0 to 10", "10 to 25", "25 to 50"]), nulls="error"
     ).name("tier2")
     expr = size.select(labelled, size[1])
 

diff --git a/ibis/backends/risingwave/tests/test_functions.py b/ibis/backends/risingwave/tests/test_functions.py
@@ -301,7 +301,7 @@ def test_category_label(alltypes, df):
     bins = [0, 10, 25, 50, 100]
     labels = ["a", "b", "c", "d"]
     bucket = d.bucket(bins)
-    expr = bucket.label(labels)
+    expr = bucket.cases(*enumerate(labels), else_=None)
     result = expr.execute()
 
     with warnings.catch_warnings():