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)