|
1 | 1 | from datetime import datetime
|
2 | 2 | import os
|
3 | 3 | import platform
|
| 4 | +from unittest import mock |
4 | 5 |
|
5 | 6 | import numpy as np
|
6 | 7 | import pytest
|
|
10 | 11 | from pandas import DataFrame
|
11 | 12 | import pandas.util.testing as tm
|
12 | 13 |
|
| 14 | + |
13 | 15 | api_exceptions = pytest.importorskip("google.api_core.exceptions")
|
14 | 16 | bigquery = pytest.importorskip("google.cloud.bigquery")
|
15 | 17 | service_account = pytest.importorskip("google.oauth2.service_account")
|
@@ -90,6 +92,41 @@ def make_mixed_dataframe_v2(test_size):
|
90 | 92 | index=range(test_size))
|
91 | 93 |
|
92 | 94 |
|
| 95 | +@mock.patch('pandas_gbq.read_gbq') |
| 96 | +def test_read_gbq_with_deprecated_kwargs(mock_read_gbq): |
| 97 | + private_key = object() |
| 98 | + mock_read_gbq.return_value = DataFrame([[1.0]]) |
| 99 | + pd.read_gbq("SELECT 1", verbose=True, private_key=private_key) |
| 100 | + _, kwargs = mock_read_gbq.call_args |
| 101 | + assert kwargs["verbose"] |
| 102 | + assert kwargs["private_key"] is private_key |
| 103 | + |
| 104 | + |
| 105 | +@mock.patch('pandas_gbq.read_gbq') |
| 106 | +def test_read_gbq_without_deprecated_kwargs(mock_read_gbq): |
| 107 | + mock_read_gbq.return_value = DataFrame([[1.0]]) |
| 108 | + pd.read_gbq("SELECT 1") |
| 109 | + _, kwargs = mock_read_gbq.call_args |
| 110 | + assert "verbose" not in kwargs |
| 111 | + assert "private_key" not in kwargs |
| 112 | + |
| 113 | + |
| 114 | +@mock.patch('pandas_gbq.read_gbq') |
| 115 | +def test_read_gbq_with_new_kwargs(mock_read_gbq): |
| 116 | + mock_read_gbq.return_value = DataFrame([[1.0]]) |
| 117 | + pd.read_gbq("SELECT 1", use_bqstorage_api=True) |
| 118 | + _, kwargs = mock_read_gbq.call_args |
| 119 | + assert kwargs["use_bqstorage_api"] |
| 120 | + |
| 121 | + |
| 122 | +@mock.patch('pandas_gbq.read_gbq') |
| 123 | +def test_read_gbq_without_new_kwargs(mock_read_gbq): |
| 124 | + mock_read_gbq.return_value = DataFrame([[1.0]]) |
| 125 | + pd.read_gbq("SELECT 1") |
| 126 | + _, kwargs = mock_read_gbq.call_args |
| 127 | + assert "use_bqstorage_api" not in kwargs |
| 128 | + |
| 129 | + |
93 | 130 | @pytest.mark.single
|
94 | 131 | class TestToGBQIntegrationWithServiceAccountKeyPath(object):
|
95 | 132 |
|
|
0 commit comments