
    'Xh                     D    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
y)    N)models)calculate_distance)calculate_multi_distance)calculate_distance_sparsec                     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 y )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         R/RAG/venv/lib/python3.12/site-packages/qdrant_client/local/tests/test_distances.pytest_distancesr&   	   s   HH_%Ehh9:G;;)%&//:M:MNQUW[P\]]];;)%&//:P:PQTWY\S]^^^;;)%&//:S:STWZ\_V`aaa;;)%&//:P:PQTWY\S]^^^HH_%Ehh9:G;;5'6??+>+>?#sRX   ;;5'6??+A+AB	Y   ;;5'6??+D+DE	c
   ;;5'6??+A+AB	S   &&1vq!fEL))2r(Aq6JKN$\>BrzzRXGYFZZZZ 	QFAq6:Ii@N ;;!,?$QW   )Y!78HHi+,-D#$5tV__=P=PQRSTX\\\\    )returnN)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 J0]r'   