
    ug                     H    d dl mZ d dlmZ d dlmZ deddfdZdeddfdZy)	    )Client)System)
invariantssqlite_persistentreturnNc                    t        j                  |       }|j                  dddd      }|j                  dddd      }||g}t        d      D ]   }|j	                  t        |      ||g       " t        d      D ]   }|j	                  t        |      ||g       " t        j                  |j                  |d	       y )
Nfirst_collection
   zhnsw:sync_thresholdzhnsw:batch_sizemetadatasecond_collection   ids
embeddingsd   T)	r   from_systemcreate_collectionrangeaddstrr   log_size_below_max_systemr   clientr	   r   collectionsis         T/var/www/openai/venv/lib/python3.12/site-packages/chromadb/test/db/test_log_purge.pytest_log_purger       s     12F//RTV%W 0  00bUW&X 1  $%67K 1XQQF;  3Z#a&aV<  !!&..+tD    c                    t        j                  |       }|j                  dddd      }|j                  dddd      }||g}t        d      D ]   }|j	                  t        |      ||g       " t        d      D ]   }|j	                  t        |      ||g       " t        j                  |j                  |d	       y )
Nr	   r
   r   r   r      r      T)	r   r   r   r   r   r   r   'log_size_for_collections_match_expectedr   r   s         r   (test_log_purge_with_multiple_collectionsr&      s     12F//RTV%W 0  00bUW&X 1  $%67K 2YQQF;  2Y#a&aV<  66Tr!   )chromadb.api.clientr   chromadb.configr   chromadb.test.propertyr   r    r&    r!   r   <module>r+      s7    & " -Ef E E. 4 r!   