test(bigquery): use range instead of getitem

ibis-project · Aug 8, 2023 · 4a066f0 · 4a066f0
1 parent 89bf919
commit 4a066f0
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/ibis/backends/bigquery/tests/system/test_client.py b/ibis/backends/bigquery/tests/system/test_client.py
@@ -9,7 +9,6 @@
 import pandas.testing as tm
 import pytest
 import pytz
-import toolz
 
 import ibis
 import ibis.expr.datatypes as dt
@@ -82,7 +81,7 @@ def test_count_distinct_with_filter(alltypes):
 
 def test_cast_string_to_date(alltypes, df):
     string_col = alltypes.date_string_col
-    month, day, year = toolz.take(3, string_col.split("/"))
+    month, day, year = map(string_col.split("/").__getitem__, range(3))
 
     expr = "20" + ibis.literal("-").join([year, month, day])
     expr = expr.cast("date")