
    _j                         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	 d dl
mZmZ d dlZ e            ededdfd                        Z e            ededdfd	                        ZdS )
    N)	ClientAPI)multi_region_test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/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/chromadb/test/distributed/test_sanity.pytest_addr1      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
             r0   r4   r4   >   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
 
 
 
 
 
r2   )r"   r!   chromadb.apir   chromadb.test.conftestr   r   r   chromadb.test.propertyr   -chromadb.test.utils.wait_for_version_increaser   r	   numpyr)   r1   r4    r2   r0   <module>r@      s:     " " " " " "         
 . - - - - -            %%	% % %  %P )y )T ) ) )  ) ) )r2   