Skip to content

Commit ba6d9b4

Browse files
committed
fixed consensus/benchmark export test
1 parent 73d3dfc commit ba6d9b4

File tree

2 files changed

+9
-9
lines changed

2 files changed

+9
-9
lines changed

tests/consensus_benchmark/test_benchmark.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,11 +25,13 @@ def test_benchmark():
2525
project_name, export_path / 'classes' / 'classes.json'
2626
)
2727
sa.upload_images_from_folder_to_project(
28-
project_name, export_path / "images"
28+
project_name, export_path / "images", annotation_status="Completed"
2929
)
3030
for i in range(1, 4):
3131
sa.upload_images_from_folder_to_project(
32-
project_name + '/consensus_' + str(i), export_path / "images"
32+
project_name + '/consensus_' + str(i),
33+
export_path / "images",
34+
annotation_status="Completed"
3335
)
3436
sa.upload_annotations_from_folder_to_project(project_name, export_path)
3537
for i in range(1, 4):

tests/consensus_benchmark/test_consensus.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -24,11 +24,13 @@ def test_consensus():
2424
project_name, export_path / 'classes' / 'classes.json'
2525
)
2626
sa.upload_images_from_folder_to_project(
27-
project_name, export_path / "images"
27+
project_name, export_path / "images", annotation_status="Completed"
2828
)
2929
for i in range(1, 4):
3030
sa.upload_images_from_folder_to_project(
31-
project_name + '/consensus_' + str(i), export_path / "images"
31+
project_name + '/consensus_' + str(i),
32+
export_path / "images",
33+
annotation_status="Completed"
3234
)
3335
sa.upload_annotations_from_folder_to_project(project_name, export_path)
3436
for i in range(1, 4):
@@ -38,11 +40,7 @@ def test_consensus():
3840
)
3941

4042
for annot_type in annot_types:
41-
res_df = sa.consensus(
42-
project_name,
43-
folder_names,
44-
annot_type=annot_type
45-
)
43+
res_df = sa.consensus(project_name, folder_names, annot_type=annot_type)
4644
#test content of projectName column
4745
assert sorted(res_df['folderName'].unique()) == folder_names
4846

0 commit comments

Comments
 (0)