Skip to content

Commit 9fa6ea4

Browse files
author
Takashi Matsuo
committed
inject the cleaner
1 parent 4e7f1d7 commit 9fa6ea4

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

datalabeling/create_annotation_spec_set_test.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -37,15 +37,16 @@ def cleaner():
3737
testing_lib.delete_annotation_spec_set(resource_name)
3838

3939

40-
def test_create_annotation_spec_set(capsys):
40+
def test_create_annotation_spec_set(cleaner, capsys):
4141

4242
@backoff.on_exception(backoff.expo, DeadlineExceeded, max_time=60)
4343
def run_sample():
4444
return create_annotation_spec_set.create_annotation_spec_set(PROJECT_ID)
4545

4646
response = run_sample()
47-
out, _ = capsys.readouterr()
48-
assert 'The annotation_spec_set resource name:' in out
4947

5048
# For cleanup
5149
cleaner.append(response.name)
50+
51+
out, _ = capsys.readouterr()
52+
assert 'The annotation_spec_set resource name:' in out

0 commit comments

Comments
 (0)