@@ -18,6 +18,8 @@ class TestAnnotationUploadVector(BaseTestCase):
1818 TEST_FOLDER_PATH = "data_set/sample_vector_annotations_with_tag_classes"
1919 TEST_4_FOLDER_PATH = "data_set/sample_project_vector"
2020 TEST_BIG_FOLDER_PATH = "sample_big_json_vector"
21+ TEST_BIG_ANNOTATION_NAME = "aearth_mov_001.jpg"
22+ TEST_BIG_ANNOTATION_PATH = os .path .join (DATA_SET_PATH , f"sample_big_json_vector/{ TEST_BIG_ANNOTATION_NAME } .json" )
2123
2224 IMAGE_NAME = "example_image_1.jpg"
2325
@@ -37,6 +39,26 @@ def _get_annotations_from_folder(path):
3739 annotations .append (json .load (open (i )))
3840 return annotations
3941
42+ def test_large_annotation_upload (self ):
43+ sa .attach_items (
44+ self .PROJECT_NAME ,
45+ [{"name" : self .TEST_BIG_ANNOTATION_NAME , "url" : f"url_" }]
46+ )
47+ sa .create_annotation_classes_from_classes_json (
48+ self .PROJECT_NAME ,
49+ f"{ self .big_annotations_folder_path } /classes/classes.json" ,
50+ )
51+ annotation = json .load (open (self .TEST_BIG_ANNOTATION_PATH ))
52+ with self .assertLogs ("sa" , level = "INFO" ) as cm :
53+ uploaded , _ , _ = sa .upload_annotations (
54+ self .PROJECT_NAME , [annotation ]
55+ ).values ()
56+ assert (
57+ "INFO:sa:Uploading 1/1 annotations to the project Test-upload_annotations."
58+ == cm .output [0 ]
59+ )
60+ assert len (uploaded ) == 1
61+
4062 def test_large_annotations_upload_get_download (self ):
4163 items_to_attach = [
4264 {"name" : f"aearth_mov_00{ i } .jpg" , "url" : f"url_{ i } " } for i in range (1 , 6 )
0 commit comments