diff --git a/bigquery/datalab-migration/samples_test.py b/bigquery/datalab-migration/samples_test.py index e178bee3ec3..50b4f2ab5c9 100644 --- a/bigquery/datalab-migration/samples_test.py +++ b/bigquery/datalab-migration/samples_test.py @@ -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')