
    	]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 ej                  d        Zd Zd Zd Zej
                  j#                  d	
      d        Zy)    N)LABEL_CONFIG_AND_TASKS)AsyncLabelStudio)LabelStudioc                     t        | |j                        }|j                  j                  dt        d         }|j                  j                  |j                  t        d          ||fS N)base_urlapi_keyzExport Test Projectlabel_config)titler
   tasks_for_importidrequest)r   r	   projectscreater   import_tasksr   django_live_urlbusiness_clientlsprojects       L/root/env/lib/python3.12/site-packages/label_studio/tests/sdk/test_export.pytest_projectr      sa    	o7N7N	OBkk  '<KabpKq rGKK

4JK]4^_w;    c                   K   t        | |j                        }|j                  j                  dt        d          d {   }|j                  j                  |j                  t        d          d {    ||fS 7 >7 
wr   )r   r	   r   r   r   r   r   r   s       r   test_project_asyncr      s     	?O<S<S	TBKK&&#2H2X '  G ++
"
"gjj:PQc:d
"
eeew;	 fs#   ?BB 5B7B8	BBc                     | \  }}|j                   j                  j                  |j                        }t	        |      dkD  sJ y )Nr   )r   exportslist_formatsr   len)r   r   r   formatss       r   test_export_formatsr"      s=    KB kk!!..wzz:Gw<!r   c           	         | \  }}|j                   j                  j                  |j                        }t	        |t
              sJ t        |      dk(  sJ |j                   j                  j                  |j                        }t	        |t        j                        sJ t        |      dk(  sJ |j                   j                  |j                  dddddddi	       |j                   j                  j                  |j                  d
      }d }t         ||            dk(  sJ |j                   j                  j                  |j                  d      }t         ||            dk(  sJ y )N   datazOpossums are greati  z2020-03-09 18:15:28.212882z
North Pole)	timestamplocation)my_textref_id	meta_infor   F)download_all_tasksc                     dd l }ddlm}  |       }| D ]  }|j                  |        |j	                  d       |j                  |      S )Nr   )BytesIO)jsonior-   writeseekload)r%   r.   r-   bufferchunks        r   _bytestream_to_jsonz/test_direct_export.<locals>._bytestream_to_json>   sD     	 ELL	 Ayy  r   T   )r   r   as_jsonr   
isinstancelistr    	as_pandaspd	DataFramer   download_sync)r   r   r   	json_datadfr%   r5   s          r   test_direct_exportr@   $   sN   KB ##++GJJ7Ii&&&y>Q 
			&	&wzz	2Bb",,'''r7a<< KK::/+GUab
  	 ;;,,WZZE,RD! "4()Q...;;,,WZZD,QD"4()Q...r   z?pytest-asyncio is not supported in this version of Label Studio)reasonc                   K   | \  }}|j                   j                  j                  |j                         d {   }t	        |t
              sJ t        |      dk(  sJ |j                   j                  j                  |j                  dddii       d {   }t	        |t        j                        sJ t        |      dk(  sJ y 7 7 3w)Nr$   task_filter_optionsfinishedonly)create_kwargs)
r   r   r7   r   r8   r9   r    r:   r;   r<   )r   r   r   r>   r?   s        r   test_async_exportrG   P   s     $KB kk))11'**==Ii&&&y>Q {{"",,WZZH]`jlr_sGt,u	uBb",,'''r7a<< >
 
vs"   8CCACC2CC)pandasr;   pytestlabel_studio.tests.sdk.commonr   mark	django_db
pytestmarklabel_studio_sdkr   label_studio_sdk.clientr   fixturer   r   r"   r@   skiprG    r   r   <module>rS      ss      @[[""
 - /  (/X Z[ \r   