feat(bigquery): implement ops.ExtractWeekOfYear

ibis-project · Jan 9, 2023 · 477d287 · 477d287
1 parent f5bacad
commit 477d287
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/ibis/backends/bigquery/registry.py b/ibis/backends/bigquery/registry.py
@@ -587,6 +587,7 @@ def _nth_value(t, op):
     ops.ExtractYear: _extract_field("year"),
     ops.ExtractQuarter: _extract_field("quarter"),
     ops.ExtractMonth: _extract_field("month"),
+    ops.ExtractWeekOfYear: _extract_field("isoweek"),
     ops.ExtractDay: _extract_field("day"),
     ops.ExtractDayOfYear: _extract_field("dayofyear"),
     ops.ExtractHour: _extract_field("hour"),

diff --git a/ibis/backends/tests/test_temporal.py b/ibis/backends/tests/test_temporal.py
@@ -126,7 +126,7 @@ def test_timestamp_extract_epoch_seconds(backend, alltypes, df):
     backend.assert_series_equal(result, expected)
 
 
-@pytest.mark.notimpl(["bigquery", "datafusion"])
+@pytest.mark.notimpl(["datafusion"])
 def test_timestamp_extract_week_of_year(backend, alltypes, df):
     expr = alltypes.timestamp_col.week_of_year().name('tmp')
     result = expr.execute()