Skip to content

fix: removes samples and tests for parameterized queries #6889

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.

Sign up for GitHub

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

Merged
merged 5 commits into from
Oct 11, 2021
Merged

fix: removes samples and tests for parameterized queries #6889

Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
efa2c5a
fix: removes samples and tests for parameterized queries
meredithslota Oct 8, 2021
ba7cc7e
Merge branch 'master' into datalab-migration-parameterized-queries
meredithslota Oct 8, 2021
507fa51
Merge branch 'master' into datalab-migration-parameterized-queries
meredithslota Oct 11, 2021
358e91c
Merge branch 'master' into datalab-migration-parameterized-queries
meredithslota Oct 11, 2021
c0d8958
Merge branch 'master' into datalab-migration-parameterized-queries
leahecole Oct 11, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
74 changes: 0 additions & 74 deletions bigquery/datalab-migration/samples_test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -155,80 +155,6 @@ def test_client_library_query_magic_results_variable(ipython_interactive):
ip.user_ns.pop(variable_name) # clean up variable


def test_datalab_magic_parameterized_query(ipython_interactive):
import pandas

ip = _set_up_ipython('google.datalab.kernel')

sample = """
# [START bigquery_migration_datalab_magic_define_parameterized_query]
%%bq query -n my_query
SELECT word, SUM(word_count) as count
FROM `bigquery-public-data.samples.shakespeare`
WHERE corpus = @corpus_name
GROUP BY word
ORDER BY count ASC
LIMIT @limit
# [END bigquery_migration_datalab_magic_define_parameterized_query]
"""
ip.run_cell(_strip_region_tags(sample))

sample = """
# [START bigquery_migration_datalab_magic_query_params]
corpus_name = "hamlet"
limit = 10
# [END bigquery_migration_datalab_magic_query_params]
"""
ip.run_cell(_strip_region_tags(sample))

sample = """
# [START bigquery_migration_datalab_magic_execute_parameterized_query]
%%bq execute -q my_query --to-dataframe
parameters:
- name: corpus_name
type: STRING
value: $corpus_name
- name: limit
type: INTEGER
value: $limit
# [END bigquery_migration_datalab_magic_execute_parameterized_query]
"""
ip.run_cell(_strip_region_tags(sample))
df = ip.user_ns["_"] # Retrieves last returned object in notebook session
assert isinstance(df, pandas.DataFrame)
assert len(df) == 10


def test_client_library_magic_parameterized_query(ipython_interactive):
import pandas

ip = _set_up_ipython('google.cloud.bigquery')

sample = """
# [START bigquery_migration_client_library_magic_query_params]
params = {"corpus_name": "hamlet", "limit": 10}
# [END bigquery_migration_client_library_magic_query_params]
"""
ip.run_cell(_strip_region_tags(sample))

sample = """
# [START bigquery_migration_client_library_magic_parameterized_query]
%%bigquery --params $params
SELECT word, SUM(word_count) as count
FROM `bigquery-public-data.samples.shakespeare`
WHERE corpus = @corpus_name
GROUP BY word
ORDER BY count ASC
LIMIT @limit
# [END bigquery_migration_client_library_magic_parameterized_query]
"""
ip.run_cell(_strip_region_tags(sample))

df = ip.user_ns["_"] # Retrieves last returned object in notebook session
assert isinstance(df, pandas.DataFrame)
assert len(df) == 10


def test_datalab_list_tables_magic(ipython_interactive):
ip = _set_up_ipython('google.datalab.kernel')

Expand Down