
    	]j/                         d dl Z d dlZd dlmZ ej                  j
                  Zd dlmZ d dl	m
Z
 d Zd Zd Zd Zd	 Zy)
    N)LABEL_CONFIG_AND_TASKS)Client)sdk_logsc                    t        | |j                        }|j                  dt        d         }dddiig}|j	                  |       |j                         }t        |      dk(  sJ |d	   d
   x}sJ |d	   d   |d	   d   k(  sJ |j                  |ddi       |j                         }t        |      dk(  sJ |d	   d   ddik(  sJ |j                  |       |j                         rJ y )Nurlapi_keyNew Projectlabel_configtitler   datamy_textz	Test task   r   idzUpdated task)r   )	r   r	   start_projectr   import_tasks	get_taskslenupdate_taskdelete_task)django_live_urlbusiness_clientlsp	task_datataskstask_ids          R/root/env/lib/python3.12/site-packages/label_studio/tests/sdk/legacy/test_tasks.pytest_task_CRUDr       s   	O_-D-D	EB
};QR`;abA9k234INN9KKMEu:??Qx~%G%&%8Fy|F3333MM'N ;M<KKMEu:??8F	>::::MM'{{}}    c           	      D   t        | |j                        }|j                  dt        d         }t	        d      D cg c]  }dddt        |      z   ii }}|j                  |       |j                         }t        |      dk(  sJ |j                  |d d	 D cg c]  }|d
   	 c}       t        |j                               d	k(  sJ |j                  |d	   d
   g       |j                         }t        |      dk(  sJ |d   d   d   dk(  sJ y c c}w c c}w )Nr   r
   r   r   
   r   r   
Test task    r   )excluded_idsr   r   zTest task 5)r   r	   r   r   rangestrr   r   r   delete_tasksdelete_all_tasks)	r   r   r   r   ir   r   tremaining_taskss	            r   test_delete_multi_tasksr.   !   s   	O_-D-D	EB
};QR`;abAGLRyQ!&9lSV&;<=QIQNN9KKMEu:NNU2AY/AdG/0q{{}"""U1Xd^$45kkmO1$$$1f%i0MAAA R 0s   DDc           	         t        | |j                        }|j                  dt        d         }t	        d      D cg c]  }dddt        |      z   ii }}|j                  |       |j                         d	   d
   }ddddddgidgi} |j                  |fi | |j                         }t        |      dk(  sJ |d	   d   d   dk(  sJ |j                  d      }t        |      dk(  sJ y c c}w )Nr   r
   r   r   r#   r   r   r$   r   r   resultlabelchoicesPositive)	from_nameto_nametypevaluer   zTest task 0T)download_all_tasks)r   r	   r   r   r'   r(   r   r   create_annotationexport_tasksr   )	r   r   r   r   r+   r   r   annotation_dataexported_taskss	            r   test_export_tasksr=   4   s   	O_-D-D	EB
};QR`;abAGLRyQ!&9lSV&;<=QIQNN9kkmAt$GY	]fisht\uvwO A3?3 ^^%N~!###!V$Y/=@@@^^t^<N~"$$$ Rs   C4c                    |j                  t        j                         t        | |j                        }|j                  dt        d         }|j                  t        d          |j                         }t        |      dk(  sJ t        |d   d         dk(  sJ t        |d   d	         dk(  sJ t        |      rJ y )
Nr   r
   r   r   tasks_for_importr   r   annotationspredictions)	set_levelloggingERRORr   r	   r   r   r   r   r   r   r   r   caplogr   r   r   s         r   1test_upload_and_list_tasks_does_not_log_to_stderrrG   J   s    
W]]#	O_-D-D	EB
};QR`;abANN)*<=>KKMEu:??uQx&'1,,,uQx&'1,,,r!   c                     |j                  t        j                         t        | |j                        }|j                  dt        d         }|j                         }|rJ t        |      rJ y )Nr   r
   r   r   )	rB   rC   rD   r   r	   r   r   r   r   rE   s         r   +test_get_empty_tasks_does_not_log_to_stderrrI   Y   sf    
W]]#	O_-D-D	EB
};QR`;abAKKME9r!   )rC   pytestlabel_studio.tests.sdk.commonr   mark	django_db
pytestmarklabel_studio_sdkr   tests.sdk.utilsr   r    r.   r=   rG   rI    r!   r   <module>rR      s=      @[[""
 # $*B&%, 	 r!   