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 run_sample():
return manage_dataset.create_dataset(PROJECT_ID)
def create_dataset(project_id):
return dataset_sample.create_dataset(project_id)
