o
    g                     @   s   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d	Ze deddfd
dZdS )    N)	ClientAPI)resetskip_if_not_cluster)
invariants)wait_for_version_increaseget_collection_versionclientreturnc                 C   s   t   }t| td| t|  | jdddddd}g }g }tdD ]"}|t| |t	j
ddd	  |jt|g|d
 gd q%t	j
ddd	 }td| tj|||d d dd|gd 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_query r-   \/var/www/visachat/venv/lib/python3.10/site-packages/chromadb/test/distributed/test_sanity.pytest_add   s:   




r/   c           	      C   s  t   }t| td| t|  | jdddddd}g }g }g }tdD ].}|t| |t	j
ddd	  |d
|  |jt|g|d g|d gd q't| |jt| |jd t	j
ddd	 }t	j
ddd	 }td| tj|||d |dd||gd 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*   r   r   r   r+   random_query_1random_query_2r-   r-   r.   r0   <   sD   




r0   )r   r   chromadb.apir   chromadb.test.conftestr   r   chromadb.test.propertyr   -chromadb.test.utils.wait_for_version_increaser   r   numpyr&   r/   r0   r-   r-   r-   r.   <module>   s   (