Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add option to use BigQuery Storage API from IPython magics #7640

Merged
merged 5 commits into from
Apr 3, 2019

Conversation

tswast
Copy link
Contributor

@tswast tswast commented Apr 1, 2019

Construct and use a BigQuery Storage API client when either the
context.use_bqstorage_api property is set, or the --use_bqstorage_api
parameter is passed into the magic.

@tswast tswast requested review from shollyman and alixhami April 1, 2019 23:21
@tswast tswast requested a review from crwilcox as a code owner April 1, 2019 23:21
@googlebot googlebot added the cla: yes This human has signed the Contributor License Agreement. label Apr 1, 2019
@tswast
Copy link
Contributor Author

tswast commented Apr 2, 2019

BigQuery tests are passing. Other failures appear to be Kokoro-induced flakes.

bigquery/google/cloud/bigquery/magics.py Outdated Show resolved Hide resolved
@pytest.mark.skipIf(
bigquery_storage_v1beta1 is None, "Requires `google-cloud-bigquery-storage`"
)
def test_bigquery_magic_with_bqstorage(monkeypatch):
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can you add a test or update an existing test asserting that the BQ storage API is not used when the new parameter is not set?

bigquery/tests/unit/test_magics.py Show resolved Hide resolved
bigquery/google/cloud/bigquery/magics.py Outdated Show resolved Hide resolved
tswast added 2 commits April 2, 2019 14:53
Construct and use a BigQuery Storage API client when either the
context.use_bqstorage_api property is set, or the --use_bqstorage_api
parameter is passed into the magic.
@tswast tswast force-pushed the bqstorage-magics branch from 99d3515 to ef4aa5d Compare April 2, 2019 21:53
@tswast tswast requested a review from alixhami April 3, 2019 22:53
@tswast tswast merged commit 33fcc1c into googleapis:master Apr 3, 2019
@tswast tswast deleted the bqstorage-magics branch April 3, 2019 23:27
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cla: yes This human has signed the Contributor License Agreement.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants