o
    3IhB                     @   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-   e/home/air/sanwanet/gpt-api/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}t| |j}g }g }g }tdD ].}|	t
| |	tjddd	  |	d
|  |jt
|g|d g|d gd q-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*   initial_versionr   r   r   r+   random_query_1random_query_2r-   r-   r.   r0   <   sF   




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   (