fix tests by using pytest raise asserts

Author: Matti Remes

pandas_gbq/tests/test_gbq.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 
 import pytest
+import warnings
 
 import re
 from datetime import datetime
@@ -174,7 +175,7 @@ def make_mixed_dataframe_v2(test_size):
 
 def test_generate_bq_schema_deprecated():
     # 11121 Deprecation of generate_bq_schema
-    with tm.assert_produces_warning(FutureWarning):
+    with pytest.warns(FutureWarning):
         df = make_mixed_dataframe_v2(10)
         gbq.generate_bq_schema(df)
 
@@ -1447,7 +1448,7 @@ def test_upload_data_with_invalid_user_schema_raises_error(self):
                        {'name': 'C', 'type': 'FLOAT'},
                        {'name': 'D', 'type': 'FLOAT'}]
         destination_table = self.destination_table + test_id
-        with tm.assertRaises(gbq.GenericGBQException):
+        with pytest.raises(gbq.GenericGBQException):
             gbq.to_gbq(df, destination_table, _get_project_id(),
                        private_key=_get_private_key_path(),
                        table_schema=test_schema)
@@ -1459,7 +1460,7 @@ def test_upload_data_with_missing_schema_fields_raises_error(self):
                        {'name': 'B', 'type': 'FLOAT'},
                        {'name': 'C', 'type': 'FLOAT'}]
         destination_table = self.destination_table + test_id
-        with tm.assertRaises(gbq.GenericGBQException):
+        with pytest.raises(gbq.GenericGBQException):
             gbq.to_gbq(df, destination_table, _get_project_id(),
                        private_key=_get_private_key_path(),
                        table_schema=test_schema)