Skip to content

Commit 3b2bae5

Browse files
committed
Fix tests to use pytest.raises
1 parent ad33b43 commit 3b2bae5

File tree

1 file changed

+18
-11
lines changed

1 file changed

+18
-11
lines changed

fast_hdbscan/tests/test_hdbscan.py

Lines changed: 18 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -132,17 +132,24 @@ def test_hdbscan_input_lists():
132132

133133

134134
def test_hdbscan_badargs():
135-
assert_raises(ValueError, fast_hdbscan, "fail")
136-
assert_raises(ValueError, fast_hdbscan, None)
137-
assert_raises(ValueError, fast_hdbscan, X, min_cluster_size="fail")
138-
assert_raises(ValueError, fast_hdbscan, X, min_samples="fail")
139-
assert_raises(ValueError, fast_hdbscan, X, min_samples=-1)
140-
assert_raises(ValueError, fast_hdbscan, X, cluster_selection_epsilon="fail")
141-
assert_raises(ValueError, fast_hdbscan, X, cluster_selection_epsilon=-1)
142-
assert_raises(ValueError, fast_hdbscan, X, cluster_selection_epsilon=-0.1)
143-
assert_raises(
144-
ValueError, fast_hdbscan, X, cluster_selection_method="fail"
145-
)
135+
with pytest.raises(ValueError):
136+
fast_hdbscan("fail")
137+
with pytest.raises(ValueError):
138+
fast_hdbscan(None)
139+
with pytest.raises(ValueError):
140+
fast_hdbscan(X, min_cluster_size="fail")
141+
with pytest.raises(ValueError):
142+
fast_hdbscan(X, min_samples="fail")
143+
with pytest.raises(ValueError):
144+
fast_hdbscan(X, min_samples=-1)
145+
with pytest.raises(ValueError):
146+
fast_hdbscan(X, cluster_selection_epsilon="fail")
147+
with pytest.raises(ValueError):
148+
fast_hdbscan(X, cluster_selection_epsilon=-1)
149+
with pytest.raises(ValueError):
150+
fast_hdbscan(X, cluster_selection_epsilon=-0.1)
151+
with pytest.raises(ValueError):
152+
fast_hdbscan(X, cluster_selection_method="fail")
146153

147154

148155
def test_fhdbscan_allow_single_cluster_with_epsilon():

0 commit comments

Comments
 (0)