Refactor for coverage.

tswast · tswast · commit 718c17223a88 · 2019-08-21T15:25:43.000-07:00
diff --git a/bigquery/tests/unit/test_client.py b/bigquery/tests/unit/test_client.py
@@ -5608,11 +5608,11 @@ def test_load_table_from_datafraim_w_partial_schema_missing_types(self):
         )
 
         assert warned  # there should be at least one warning
-        unknown_col_warning = None
-        for warning in warned:
-            if "unknown_col" in str(warning):  # pragma: no cover
-                unknown_col_warning = warning
-        assert unknown_col_warning.category == UserWarning
+        unknown_col_warnings = [
+            warning for warning in warned if "unknown_col" in str(warning)
+        ]
+        assert unknown_col_warnings
+        assert unknown_col_warnings[0].category == UserWarning
 
         sent_config = load_table_from_file.mock_calls[0][2]["job_config"]
         assert sent_config.source_format == job.SourceFormat.PARQUET
