diff --git a/tests/boundary/test_bigquery_sdk.py b/tests/boundary/test_bigquery_sdk.py index 52e8db311..b8e6c9995 100644 --- a/tests/boundary/test_bigquery_sdk.py +++ b/tests/boundary/test_bigquery_sdk.py @@ -5,16 +5,14 @@ from google.api_core.exceptions import NotFound -class TestBigQuerySDK: +@pytest.mark.parametrize("table_name", ["this_table_does_not_exist"]) +def test_get_table_does_not_exist(project, table_name): """ TODO: replace dbt project methods with direct connection instantiation """ - - @pytest.mark.parametrize("table_name", ["this_table_does_not_exist"]) - def test_get_table_does_not_exist(self, project, table_name): - with get_connection(project.adapter) as conn: - client: Client = conn.handle - dataset_ref = DatasetReference(project.database, project.test_schema) - table_ref = TableReference(dataset_ref, table_name) - with pytest.raises(NotFound): - client.get_table(table_ref) + with get_connection(project.adapter) as conn: + client: Client = conn.handle + dataset_ref = DatasetReference(project.database, project.test_schema) + table_ref = TableReference(dataset_ref, table_name) + with pytest.raises(NotFound): + client.get_table(table_ref)