
    <i                     F    S SK rS SKJr  S SKJr  S SKJr  S SKJ	r	  SS jr
g)    N)models)calculate_distance)calculate_multi_distance)calculate_distance_sparsec                     [         R                  " / SQ5      n [         R                  " / SQ/ SQ/5      n[         R                  " [        X[        R
                  R                  5      SS/5      (       d   e[         R                  " [        X[        R
                  R                  5      SS/5      (       d   e[         R                  " [        X[        R
                  R                  5      SS/5      (       d   e[         R                  " [        X[        R
                  R                  5      SS/5      (       d   e[         R                  " / SQ5      n [         R                  " / SQ/ SQ/5      n[         R                  " [        X[        R
                  R                  5      SS/SS	9(       d   e[         R                  " [        X[        R
                  R                  5      S
S/SS	9(       d   e[         R                  " [        X[        R
                  R                  5      SS/SS	9(       d   e[         R                  " [        X[        R
                  R                  5      SS/SS	9(       d   e[        R                  " SS/SS/S9n[        R                  " SS/SS/S9/n[        X#5      [         R                  " S5      /:X  d   e[        R                  " SS/SS/S9[        R                  " / SQ/ SQS9/n[         R                  " [        X#5      SS/SS	9(       d   e[         R                  " / SQ/ SQ/5      n[         R                  " / SQ/ SQ/5      /n[        XE[        R
                  R                  5      S   S:X  d   eg )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         b/home/james-whalen/.local/lib/python3.13/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a;;)%&//:P:PQTWY\S]^^^^HH_%Ehh9:G;;56??+>+>?#sRX   ;;56??+A+AB	Y   ;;56??+D+DE	c
   ;;56??+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#$5V__=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'   