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)
