Ejemplos de _test_google_api_imports en Python

Ejemplo n.º 1

Archivo: test_gbq.py Proyecto: pydata/pandas-gbq

def test_GbqConnector_get_client_w_new_bq(mock_bigquery_client):
    gbq._test_google_api_imports()
    pytest.importorskip("google.api_core.client_info")

    connector = _make_connector()
    connector.get_client()

    _, kwargs = mock_bigquery_client.call_args
    assert kwargs["client_info"].user_agent == "pandas-{}".format(pandas.__version__)

Ejemplo n.º 2

Archivo: test_gbq.py Proyecto: mech-mocha/pandas-gbq

def test_GbqConnector_get_client_w_old_bq(monkeypatch, mock_bigquery_client):
    gbq._test_google_api_imports()
    connector = _make_connector()
    monkeypatch.setattr(gbq, "HAS_CLIENT_INFO", False)

    connector.get_client()

    # No client_info argument.
    mock_bigquery_client.assert_called_with(credentials=mock.ANY,
                                            project=mock.ANY)

Ejemplo n.º 3

def test_GbqConnector_get_client_w_new_bq(mock_bigquery_client):
    gbq._test_google_api_imports()
    if not FEATURES.bigquery_has_client_info:
        pytest.skip("google-cloud-bigquery missing client_info feature")
    pytest.importorskip("google.api_core.client_info")

    connector = _make_connector()
    connector.get_client()

    _, kwargs = mock_bigquery_client.call_args
    assert kwargs["client_info"].user_agent == "pandas-{}".format(
        pandas.__version__)

Ejemplo n.º 4

def test_GbqConnector_get_client_w_old_bq(monkeypatch, mock_bigquery_client):
    gbq._test_google_api_imports()
    connector = _make_connector()
    monkeypatch.setattr(
        type(FEATURES),
        "bigquery_has_client_info",
        mock.PropertyMock(return_value=False),
    )

    connector.get_client()

    # No client_info argument.
    mock_bigquery_client.assert_called_with(credentials=mock.ANY,
                                            project=mock.ANY)

Ejemplo n.º 5

Archivo: test_gbq.py Proyecto: pydata/pandas-gbq

def test_imports():
    gbq._test_google_api_imports()