
    Ng                     F    d dl Zd dlmZ d dlmZ d dlmZ d dlm	Z	 ddZ
dS )	    N)models)calculate_distance)calculate_multi_distance)calculate_distance_sparsereturnc                  ~   t          j        g d          } t          j        g dg dg          }t          j        t          | |t          j        j                  ddg          sJ t          j        t          | |t          j        j                  ddg          sJ t          j        t          | |t          j        j                  ddg          sJ t          j        t          | |t          j        j	                  ddg          sJ t          j        g d          } t          j        g dg dg          }t          j        t          | |t          j        j                  ddgd	          sJ t          j        t          | |t          j        j                  d
dgd	          sJ t          j        t          | |t          j        j                  ddgd	          sJ t          j        t          | |t          j        j	                  ddgd	          sJ t	          j
        ddgddg          }t	          j
        ddgddg          g}t          ||          t          j        d          gk    sJ t	          j
        ddgddg          t	          j
        g dg d          g}t          j        t          ||          ddgd	          sJ t          j        g dg dg          }t          j        g dg dg          g}t          ||t          j        j                  d         dk    sJ d S )N)      ?g       @      @g      ,@        r	   )r	   r   r	   )r   r	   r   g      @g-C6?)atolg'e@g+Pz?r
   g00?      )indicesvalues
      z-inf      )r   r   r   g      &@   )r   r   r   )r   r   r   r   g      D@)nparrayallcloser   r   DistanceDOTEUCLID	MANHATTANCOSINESparseVectorr   float32r   )queryvectorssparse_querysparse_vectorsmultivector_querydocss         d/var/www/html/ai-engine/env/lib/python3.11/site-packages/qdrant_client/local/tests/test_distances.pytest_distancesr'   	   s   H___%%Eh9::G;)%&/:MNNQUW[P\]]]]];)%&/:PQQTWY\S]^^^^^;)%&/:STTWZ\_V`aaaaa;)%&/:PQQTWY\S]^^^^^H___%%Eh9::G;5'6?+>??#sRX      ;5'6?+ABB	Y      ;5'6?+DEE	c
      ;5'6?+ABB	S      &1vq!fEEEL)2r(Aq6JJJKN$\>BBrzRXGYGYFZZZZZ 	QFAq6:::IIIiii@@@N ;!,??$QW      )))YYY!788Hiii+,,-D#$5tV_=PQQRSTX\\\\\\\    )r   N)numpyr   qdrant_client.httpr   qdrant_client.local.distancesr   #qdrant_client.local.multi_distancesr   $qdrant_client.local.sparse_distancesr   r'    r(   r&   <module>r/      s}        % % % % % % < < < < < < H H H H H H J J J J J J0] 0] 0] 0] 0] 0]r(   