chore: add teardown resource fixture to system tests (#1247)

googleapis · May 20, 2022 · 0c4ead4 · 0c4ead4
1 parent a2eb8cc
commit 0c4ead4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/tests/system/aiplatform/test_dataset.py b/tests/system/aiplatform/test_dataset.py
@@ -189,6 +189,7 @@ def test_get_new_dataset_and_import(self, dataset_gapic_client):
             my_dataset.import_data(
                 gcs_source=_TEST_TEXT_ENTITY_EXTRACTION_GCS_SOURCE,
                 import_schema_uri=_TEST_TEXT_ENTITY_IMPORT_SCHEMA,
+                import_request_timeout=500,
             )
 
             data_items_post_import = dataset_gapic_client.list_data_items(

diff --git a/tests/system/aiplatform/test_featurestore.py b/tests/system/aiplatform/test_featurestore.py
@@ -51,6 +51,7 @@
     "delete_staging_bucket",
     "prepare_bigquery_dataset",
     "delete_bigquery_dataset",
+    "tear_down_resources",
 )
 class TestFeaturestore(e2e_base.TestEndToEnd):