def test_delete_dataset(capsys): # Creates a dataset. response = manage_dataset.create_dataset(PROJECT_ID) manage_dataset.delete_dataset(response.name) out, _ = capsys.readouterr() assert "Dataset deleted." in out
def test_create_dataset(capsys): response = manage_dataset.create_dataset(PROJECT_ID) out, _ = capsys.readouterr() assert "The dataset resource name:" in out # clean up manage_dataset.delete_dataset(response.name)
def dataset(): # create a temporary dataset dataset = manage_dataset.create_dataset(PROJECT_ID) yield dataset # tear down manage_dataset.delete_dataset(dataset.name)
def dataset(): # create a temporary dataset dataset = manage_dataset.create_dataset(PROJECT_ID) # import some data to it import_data.import_data(dataset.name, 'VIDEO', INPUT_GCS_URI) yield dataset # tear down manage_dataset.delete_dataset(dataset.name)
def dataset(): # create a temporary dataset dataset = manage_dataset.create_dataset(PROJECT_ID) # import some data to it import_data.import_data(dataset.name, 'IMAGE', INPUT_GCS_URI) yield dataset # tear down manage_dataset.delete_dataset(dataset.name)
def delete_old_datasets(project_id): client = create_client() formatted_project_name = client.project_path(project_id) response = client.list_datasets(formatted_project_name) # It will delete datasets created more than 2 hours ago cutoff_time = time.time() - 7200 for element in response: if element.create_time.seconds < cutoff_time: print("Deleting {}".format(element.name)) try: dataset_sample.delete_dataset(element.name) except FailedPrecondition as e: # We're always getting FailedPrecondition with 400 # resource conflict. I don't know why. print("Deleting {} failed.".format(element.name)) print("Detail: {}".format(e)) # To avoid quota error time.sleep(1)
def run_sample(): manage_dataset.delete_dataset(dataset.name)
def delete_dataset(name): return dataset_sample.delete_dataset(name)