fix: removes samples and tests for parameterized queries (GoogleCloudPlatform#6889)

meredithslota · leahecole · web-flow · commit 00dc7dedb112 · 2021-10-11T15:50:47.000-07:00
Co-authored-by: Leah E. Cole &lt;6719667+leahecole@users.noreply.github.com&gt;
diff --git 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')
 
