@@ -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