Skip to content

Commit 8643755

Browse files
committed
fix: remove pytest import in prod code (#957)
1 parent 6cb0a60 commit 8643755

File tree

1 file changed

+10
-11
lines changed

1 file changed

+10
-11
lines changed

qdrant_client/hybrid/formula.py

Lines changed: 10 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import math
2-
import pytest
2+
33

44
from qdrant_client.conversions.common_types import get_args_subscribed
55
from qdrant_client.http import models
@@ -358,14 +358,13 @@ def test_parsing_variable() -> None:
358358
assert str(e) == "Invalid score pattern: $score[10].other"
359359

360360

361-
@pytest.mark.parametrize(
362-
"payload_value, expected", [(1.2, 1.2), ([1.2], 1.2), ([1.2, 2.3], [1.2, 2.3])]
363-
)
364-
def test_try_extract_payload_value(payload_value: Any, expected: Any) -> None:
365-
empty_defaults: dict[str, Any] = {}
366-
payload = {"key": payload_value}
367-
assert try_extract_payload_value("key", payload, empty_defaults) == expected
361+
def test_try_extract_payload_value() -> None:
362+
for payload_value, expected in [(1.2, 1.2), ([1.2], 1.2), ([1.2, 2.3], [1.2, 2.3])]:
363+
empty_defaults: dict[str, Any] = {}
364+
365+
payload = {"key": payload_value}
366+
assert try_extract_payload_value("key", payload, empty_defaults) == expected
368367

369-
defaults = {"key": payload_value}
370-
empty_payload: dict[str, Any] = {}
371-
assert try_extract_payload_value("key", empty_payload, defaults) == expected
368+
defaults = {"key": payload_value}
369+
empty_payload: dict[str, Any] = {}
370+
assert try_extract_payload_value("key", empty_payload, defaults) == expected

0 commit comments

Comments
 (0)