Skip to content

Commit 5a14ce2

Browse files
committed
updates an import step and restores gc import
1 parent 1b8ff0b commit 5a14ce2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/unit/test__pandas_helpers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
import datetime
1717
import decimal
1818
import functools
19+
import gc
1920
import operator
2021
import queue
2122
from typing import Union
@@ -1886,7 +1887,6 @@ def fake_download_stream(
18861887
assert queue_used.maxsize == expected_maxsize
18871888

18881889

1889-
'''
18901890
@pytest.mark.skipif(isinstance(pyarrow, mock.Mock), reason="Requires `pyarrow`")
18911891
def test__download_table_bqstorage_shuts_down_workers(
18921892
monkeypatch,
@@ -1897,6 +1897,7 @@ def test__download_table_bqstorage_shuts_down_workers(
18971897
Make sure that when the top-level iterator goes out of scope (is deleted),
18981898
the child threads are also stopped.
18991899
"""
1900+
pytest.importorskip("google.cloud.bigquery_storage_v1")
19001901
from google.cloud.bigquery import dataset
19011902
from google.cloud.bigquery import table
19021903
import google.cloud.bigquery_storage_v1.reader
@@ -2013,7 +2014,6 @@ def test_download_arrow_row_iterator_unknown_field_type(module_under_test):
20132014
col = result.columns[1]
20142015
assert type(col) is pyarrow.lib.DoubleArray
20152016
assert col.to_pylist() == [2.2, 22.22, 222.222]
2016-
'''
20172017

20182018

20192019
@pytest.mark.skipif(isinstance(pyarrow, mock.Mock), reason="Requires `pyarrow`")

0 commit comments

Comments
 (0)