Skip to content

Commit a2bd4ef

Browse files
committed
Rename parameters for readability
1 parent d32a991 commit a2bd4ef

File tree

1 file changed

+24
-16
lines changed

1 file changed

+24
-16
lines changed

tests/distances/utils.py

Lines changed: 24 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -117,19 +117,27 @@ def run_test_bulk_similarity_and_distance_two_arrays(
117117
assert np.allclose(bulk_dist, bulk_dist_sparse, atol=1e-3)
118118

119119

120-
def assert_similarity_values(similarity: float, comparison: str, value: float) -> None:
121-
if comparison == ">":
122-
assert similarity > value
123-
elif comparison == "<":
124-
assert similarity < value
125-
elif comparison == "==":
126-
assert np.isclose(similarity, value, atol=1e-3)
127-
128-
129-
def assert_distance_values(distance: float, comparison: str, value: float) -> None:
130-
if comparison == ">":
131-
assert value > distance
132-
elif comparison == "<":
133-
assert value < distance
134-
elif comparison == "==":
135-
assert np.isclose(distance, value, atol=1e-3)
120+
def assert_similarity_values(
121+
similarity: float,
122+
comparison_sign: str,
123+
threshold: float,
124+
) -> None:
125+
if comparison_sign == ">":
126+
assert similarity > threshold
127+
elif comparison_sign == "<":
128+
assert similarity < threshold
129+
elif comparison_sign == "==":
130+
assert np.isclose(similarity, threshold, atol=1e-3)
131+
132+
133+
def assert_distance_values(
134+
distance: float,
135+
comparison_sign: str,
136+
threshold: float,
137+
) -> None:
138+
if comparison_sign == ">":
139+
assert threshold > distance
140+
elif comparison_sign == "<":
141+
assert threshold < distance
142+
elif comparison_sign == "==":
143+
assert np.isclose(distance, threshold, atol=1e-3)

0 commit comments

Comments
 (0)