
    LKiB                         S SK r S SKrS SKJr  S SKJrJr  S SKJr  S SK	J
r
Jr  S SKr\" 5       S\SS4S j5       r\" 5       S\SS4S	 j5       rg)
    N)	ClientAPI)resetskip_if_not_cluster)
invariants)wait_for_version_increaseget_collection_versionclientreturnc                 H   [         R                   " 5       n[        R                  " U5        [        SU5        [	        U 5        U R                  SSSSS.S9n/ n/ n[        S5       Hm  nUR                  [        U5      5        UR                  [        R                  R                  SS5      S	   5        UR                  [        U5      /US
   /S9  Mo     [        R                  R                  SS5      S	   n[        SU5        [        R                  " UUUS S S.SU/S9  g )NGenerating data with seed test   zhnsw:construction_efzhnsw:search_efzhnsw:Mnamemetadata        r   )ids
embeddingsGenerated data with seed r   r   	metadatas	documents
   query_embeddings)timerandomseedprintr   create_collectionrangeappendstrnprandaddr   ann_accuracy)r	   r"   
collectionr   r   irandom_querys          g/var/www/html/dynamic-report/venv/lib/python3.13/site-packages/chromadb/test/distributed/test_sanity.pytest_addr0      s    99;D
KK	
&-	&M))*-PST * J CJ4[

3q6"))..A.q12Q"2' 	 	
  99>>!Q'*L	
%t,$		
 	&
    c                    [         R                   " 5       n[        R                  " U5        [        SU5        [	        U 5        U R                  SSSSS.S9n[        XR                  5      n/ n/ n/ n[        S5       H  nUR                  [        U5      5        UR                  [        R                  R                  SS5      S	   5        UR                  S
U 35        UR                  [        U5      /US   /US   /S9  M     [        XR                  US5        [        R                  R                  SS5      S	   n[        R                  R                  SS5      S	   n	[        SU5        [        R                   " UUUS US.SX/S9  g )Nr   *test_add_include_all_with_compaction_delayr   r   r   r   r   r   r   	document_r   )r   r   r   x   r   r   r   r   )r    r!   r"   r#   r   r$   r   r   r%   r&   r'   r(   r)   r*   r   r   r+   )
r	   r"   r,   initial_versionr   r   r   r-   random_query_1random_query_2s
             r/   r3   r3   <   sl   99;D
KK	
&-	&M))9*-PST * J -V__EO
CJI4[

3q6"))..A.q129QC)Q"2' }o 	 	
	  fooLYY^^Aq)!,NYY^^Aq)!,N	
%t, $"		
 	(9
r1   )r!   r    chromadb.apir   chromadb.test.conftestr   r   chromadb.test.propertyr   -chromadb.test.utils.wait_for_version_increaser   r   numpyr(   r0   r3    r1   r/   <module>r?      sq      " .  %%	% %P )y )T ) )r1   