
    ug                         d dl Z d dlZd dlmZ d dlmZmZ d dlmZ d dl	m
Z
mZ d dlZ e       deddfd       Z e       deddfd	       Zy)
    N)	ClientAPI)resetskip_if_not_cluster)
invariants)wait_for_version_increaseget_collection_versionclientreturnc                 P   t        j                          }t        j                  |       t        d|       t	        |        | j                  ddddd      }g }g }t        d      D ]o  }|j                  t        |             |j                  t        j                  j                  dd      d	          |j                  t        |      g|d
   g       q t        j                  j                  dd      d	   }t        d|       t        j                  |||d d dd|g       y )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          Z/var/www/openai/venv/lib/python3.12/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                 &   t        j                          }t        j                  |       t        d|       t	        |        | j                  ddddd      }g }g }g }t        d      D ]  }|j                  t        |             |j                  t        j                  j                  dd      d	          |j                  d
|        |j                  t        |      g|d   g|d   g        t        | |j                  t        | |j                        d       t        j                  j                  dd      d	   }t        j                  j                  dd      d	   }t        d|       t        j                   |||d |dd||g       y )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,   r   r   r   r-   random_query_1random_query_2s	            r/   r3   r3   <   sr   99;D
KK	
&-	&M))9*-PST * J
 CJI4[

3q6"))..A.q129QC)Q"2' }o 	 	
	  fjoo7MfV`VeVe7fhk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   