codekingpro/portable-devtools
114k
1�
2 ��j] � � � S SK r S SKrS SKJrJr S SKrS SKrS SKrS SKJ r S SK3J r J r " S S\ R 5 r " S S\ R 5 rS4S jrg) � N)�patch�Mock)�SysDB)�Consumer�Producerc � � \ rS rSr\" SSS9S\SS4S j5 r\" SSS9S\SS4S j5 r\" SSS9S\SS4S5 j5 rS r g) � GetDBTest� � chromadb.db.impl.sqlite.SqliteDBT��autospec�mock�returnNc �� � [ R R [ R R SS95 nUR [6 5 UR ( d eg �N�./foo)�persist_directory��chromadb�config�System�Settings�instancer �called��selfr �systems �_D:\code\apps\devtools\python\user_packages\Python313\site-packages\chromadb/test/test_chroma.py�test_default_db�GetDBTest.test_default_db sE � �� � �'�'� �O�O� $� $�w� $� ?�7�� ���� ��{�{��{� c �� � [ R R [ R R SSS95 nUR [8 5 UR ( d eg )Nr r )�chroma_sysdb_implr r r s r �test_sqlite_sysdb�GetDBTest.test_sqlite_sysdb sM � �� � �'�'� �O�O� $� $�"D�")�
%�
�9�� ���� ��{�{��{r! c �� � [ R R [ R R SSSSS95 nUR [10 5 UR [ 5 UR ( d eg )Nr r )r# �chroma_producer_impl�chroma_consumer_implr )r r r r r r r r r s r �test_sqlite_queue�GetDBTest.test_sqlite_queue s_ � �� � �'�'� �O�O� $� $�"D�%G�%G�")�
%�
�11�� ����!�����!��{�{��{r! � )12�__name__�13__module__� __qualname__�__firstlineno__r r r r$ r) �__static_attributes__r+ r! r r r s~ � �14� -��=��D� �T� � >�� � -��=��d� �t� � >�� � -��=� �d� �t� � >� r! r c � � \ rS rSr\" SSS9\R15 " \R SS0SS9S\SS 4S16 j5 5 r \" S SS9\R17 " \R SS0SS9S \SS 4S
j5 5 r18\" SSS9\R19 " \R 0 SS9S\SS 4S j5 5 r \" SSS9\R20 " \R 0 SS9S\SS 4S j5 5 r Sr
g )�21GetAPITest�. z chromadb.api.segment.SegmentAPITr �CHROMA_API_IMPL)�clear�mock_apir Nc � � [ R " [ R R SS95 nUR ( d eUR 5 g r �r �Clientr r r �clear_system_cache)r r6 �clients r �22test_local�GetAPITest.test_local/ s< � �23 � � �� � �!9�!9�G�!9�!T�U����� ���!�!�#r! r �mock_dbc � � [ R " [ R R SS95 nUR ( d eUR 5 g r r8 )r r> r; s r �
test_local_db�GetAPITest.test_local_db8 s< � �24 � � �� � �!9�!9�G�!9�!T�U���~�~� �~��!�!�#r! � chromadb.api.fastapi.FastAPIr c � � [ R " [ R R SSSSS95 nUR ( d eUR 5 g )NrB r �foo�P ��chroma_api_implr �chroma_server_host�chroma_server_http_portr8 )r r r; s r � test_fastapi�GetAPITest.test_fastapiA sO � � � � � �O�O� $� $� >�")�#(�(*�
%�
�25�� �{�{��{��!�!�#r! c � � [ R R SSSSS0S9n[ R " U5 nUR ( d eUR26 u pEU( a US OS nUR U:X d eUR 5 g )NrB rD rE �bar)rG rH rI �chroma_server_headersr )r r r r9 r � call_args�settingsr: )r r rP r; �args�kwargs�passed_settingss r � test_settings_pass_to_fastapi�(GetAPITest.test_settings_pass_to_fastapiO s� � � �?�?�+�+� :� $�$&�#(�%�.� ,� 27�� � � ��*�� �{�{��{� �~�~� ��%)�$�q�'�t�� �'�'�8�3�3�3��!�!�#r! r+ )r, r- r. r/ r �dict�os�environr r< r@ rJ rT r0 r+ r! r r2 r2 . s" � �28� ,�t�<�29�Z�Z�30�31�32�&�(I�J�RV��$�4� $�D� $�� =�$�33 � -��=�34�Z�Z�35�36�37�&�(I�J�RV��$�T� $�d� $�� >�$�38 � )�D�9�39�Z�Z��40�41�B�d�+�42$�� 43$�$� 44$� ,� :�45$� � )�D�9�46�Z�Z��47�48�B�d�+�$�$� $�4� $� ,� :�$r! r2 c 49 �� � [ R " [ 5 [ R " [ R50 R
SSSSS95 n U R 5 S S S 5 g ! , ( d f g = f)Nzchromadb.api.local.LocalAPIr rD rE rF )�pytest�raises�51ValueErrorr r9 r r r: )r; s r �test_legacy_valuesr] j sY � � � � �z� "�� � � �O�O� $� $� =�")�#(�(*�
%�
�52�� �!�!�#� 53#� "� "�s �AA)�)54A7)r N)�unittestrW �
unittest.mockr r rZ r �chromadb.config�chromadb.db.systemr �chromadb.ingestr r �TestCaser r2 r] r+ r! r �<module>rd sH �� � � %�
� � � $� .� ��!�!� �F9$��"�"� 9$�x55$r!