
    	]j+                     *   d Z ddlZddlZddlmZ ddlmZmZmZ ddl	m
Z
 ej                  j                  d      ej                  j                  dd	dd
ddddgidddgdddddgdfddd
ddddgidddgdddddddd
ddddgidddgdddddgdfddd
ddddgidddgdddddddd
ddddgidddgdddddddd
ddddgidddgdddddgdfddd
ddddgidddgdddddgdfg      ej                  j                  dg d      ej                  j                  d                              Zej                  j                  d      ej                  j                  d!d"      ej                  j                  d#d"      ej                  j                  d$d%      ej                  j                  d&dddddd'gid(ggdddddd'gid(gd)d*fdddddd'gid(ggdfg      ej                  j                  d+                                           Zy),zThis file and its contents are licensed under the Apache License 2.0. Please see the included NOTICE for copyright information and LICENSE for a copy of the license.
    N)apps)
Annotation
PredictionTask)AnnotationSerializerzHTX-868)reasonz"annotation_items, aggregated_class   F123choicesclass_AAtext
text_class)idtypevalueto_name	from_nametest   )r   review_resultground_truthresult
created_at
updated_at	lead_timecompleted_by       456      class_BB789   
   z:finished, aggregator_type, return_task, num_task_in_result))0no_aggregationr'   r   )1r(   r'   r	   )r'   majority_voter'   r   )r'   r*   r)   r   )r)   r*   r)   r	   c                    |dk(  r*t        j                  d      st        j                  d       t        j
                  j                  |j                        }|j                         t               }	|D ]  }
t        j
                  j                  |
d   | j                        }t        |      j                  }|d   |
d<   |d   |
d<   |d	   |
d	<   | j                  j                  |
d
<   |	j                  t!        j"                  |
              | j%                  d|j                   d|||d      }|j&                  dk(  sJ |j!                         }t)        |      |k(  sJ |dk(  r(t+        fd|D              }|d   j                  k(  sFJ |dk(  r(t+        fd|D              }|d   j                  k(  sJ t-        dt/        |      z         |dk(  rzt               }|d   D ]&  }
|j                  t!        j"                  |
             ( ||	k(  sJ |dk7  r5t)        |      |j1                         k(  sJ t3        fd|D              sJ y y |d   d   d   d   d   d   d   |k(  sJ y )Nr*   
businesses1Not supported aggregation for open-source versionprojectr   taskr   r   r   r   r   r   /api/projects/	/results/)finishedaggregator_typereturn_taskdata   r'   c              3   H   K   | ]  }|d    j                   k(  s|  ywr   Nr   .0tr1   s     H/root/env/lib/python3.12/site-packages/label_studio/tests/test_export.py	<genexpr>ztest_export.<locals>.<genexpr>         $N11T7dgg;MQ$N   ""r)   c              3   H   K   | ]  }|d    j                   k(  s|  ywr;   r<   r=   s     r@   rA   ztest_export.<locals>.<genexpr>   rB   rC   r8   z%Incorrect return_task param in test: r(   annotationsc              3   f   K   | ](  }|d    j                   k7  st        |d         dk(   * yw)r   rE   r   N)r   lenr=   s     r@   rA   ztest_export.<locals>.<genexpr>   s2     YaagQUQXQXFXs1]+,1Ys   11r   r   r   )r   is_installedpytestskipr   objectsfilterr   firstsetr   createadminr   r8   addjsondumpsgetstatus_coderG   next	Exceptionstrcountall)business_clientconfigured_projectr4   r5   r6   num_task_in_resultannotation_itemsaggregated_class
task_queryexpected_annotations_for_task
annotationdb_annotationrexportstask_with_annotationexported_annotationsr1   s                   @r@   test_exportrh      s   Z /)$2C2CL2QGH$$-?-B-B$CJD$'E!& 	B
"**11j2AVAV 2 
 -];@@(.
4#0#>
< #0#>
< %4%:%:%=%=
>"%))$**Z*@A	B 	
+../y9"Wbc 	 	A ==CffhG w<---- c#$N$NO#D)TWW444		#$N$NO#F+tyy888?#kBRRSS **"u.}= 	=J $$TZZ
%;<	=#'DDDDs?w<:#3#3#5555Y7YYYY	  $M215h?B7KIVWXY]mmmm    r4   )r'   r)   r6   r5   )r(   r*   zannotation_results, predictionsclass_A)r   r   r   r   r   g      ?)r   scorec                    |dk(  r*t        j                  d      st        j                  d       t        j
                  j                  |j                        }|j                         }|D ]O  }	|	D ]  }
| j                  j                  g|
d<    t        j
                  j                  ||	| j                         Q |r:|D ]5  }t        j
                  j                  ||j                  |d   |d   	       7 | j                  d
|j                   d|||dd      }
|
j                  dk(  sJ |
j!                         }|D ]5  }|r'|d   d   d   |d   k(  sJ |d   d   d   |d   k(  r*J |d   g k(  r5J  y )Nr*   r,   r-   r.   r   r0   r   rk   )r1   r/   r   rk   r2   r3   r)   )r4   r5   r6   return_predictionsr7   r9   predictionsr   )r   rH   rI   rJ   r   rK   rL   r   rM   rP   r   rO   r   r/   rT   rU   rR   )r[   r\   r4   r6   r5   annotation_resultsrn   tasksr1   r   rd   re   s               r@   test_export_with_predictionsrq      s   h /)$2C2CL2QGHLL(:(=(=>E;;=D$ ` 	;A!0!6!6!9!9 :An	;!!tFI^I^!_`  	D%%4<<H8MU`ahUi & 	
 	
+../y9 .&"%	
 	 	A ==CffhG -&q)(3{87LLLL&q)'2k'6JJJJ&",,,-ri   )__doc__rR   rI   django.appsr   tasks.modelsr   r   r   tasks.serializersr   markrJ   parametrize	django_dbrh   rq    ri   r@   <module>rz      s      5 5 2 #(
 %)$) #($-&/*%>'-)5 #)"(!%$%!( +	
4 %)$) #($-&/*%>'-)5 #%"$!%$%!& %)$) #($-&/*%>'-)5 #%"$!%$%!'%L O(	
X %)$) #($-&/*%>'-)5 #%"$!%$%!& %)$) #($-&/*%>'-)5 #%"$!%$%!& %)$) #($-&/*%>'-)5 #%"$!%$%!K7p s:	
| %)$) #($-&/*%>'-)5 #%"$!%$&!( +	
ySVn @	 <n 	oV $F<n~ #Z0
3*,OP% $%1#) )"+i[!9
 $%1#) )"+i[!9 	
: $%1#) )"+i[!9
 	
5(+X "- Y+ Q 4 1 $b"-ri   