
    ZiB                         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dS )
    N)	ClientAPI)resetskip_if_not_cluster)
invariants)wait_for_version_increaseget_collection_versionclientreturnc                    t          j                     }t          j        |           t          d|           t	          |            |                     ddddd          }g }g }t          d          D ]}|                    t          |                     |                    t          j        
                    dd          d	                    |                    t          |          g|d
         g           t          j        
                    dd          d	         }t          d|           t          j        |||d d dd|g           d S )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          b/root/projects/butler/venv/lib64/python3.11/site-packages/chromadb/test/distributed/test_sanity.pytest_addr0      sq    9;;D
K	
&---	&MMM))*-PSTT *  J CJ4[[ 
 


3q66")..A..q1222Q"2' 	 	
 	
 	
 	

 9>>!Q''*L	
%t,,,$		
 	
 	&
 
 
 
 
 
    c                    t          j                     }t          j        |           t          d|           t	          |            |                     ddddd          }t          | |j                  }g }g }g }t          d          D ]}|	                    t          |                     |	                    t          j                            dd          d	                    |	                    d
|            |                    t          |          g|d         g|d         g           t          | |j        |d           t          j                            dd          d	         }t          j                            dd          d	         }	t          d|           t          j        |||d |dd||	g           d S )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   <   s   9;;D
K	
&---	&MMM))9*-PSTT *  J -VZ_EEO
CJI4[[ 
 


3q66")..A..q1222Q)))Q"2' }o 	 	
 	
 	
 	
 fjoLLLY^^Aq))!,NY^^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?      s     " " " " " "        . - - - - -            %%	% % % %P )y )T ) ) ) ) ) )r1   