diff --git a/tests/unit/test_table.py b/tests/unit/test_table.py index 2a49b0632..7a97c7b78 100644 --- a/tests/unit/test_table.py +++ b/tests/unit/test_table.py @@ -2362,13 +2362,14 @@ def test__should_use_bqstorage_returns_false_if_max_results_set(self): def test__should_use_bqstorage_returns_false_w_warning_if_missing_dependency(self): iterator = self._make_one(first_page_response=None) # not cached - + def fail_bqstorage_import(name, globals, locals, fromlist, level): """Returns True if bigquery_storage has been imported.""" # NOTE: *very* simplified, assuming a straightforward absolute import return "bigquery_storage" in name or ( fromlist is not None and "bigquery_storage" in fromlist ) + # maybe_fail_import() returns ImportError if the predicate is True no_bqstorage = maybe_fail_import(predicate=fail_bqstorage_import)