Skip to content

Commit 1e2c408

Browse files
committed
update check_estimator to parametrize_with_checks in structural tests
1 parent b7aa96c commit 1e2c408

File tree

2 files changed

+9
-10
lines changed

2 files changed

+9
-10
lines changed

.coverage

0 Bytes
Binary file not shown.

tests/test_structural.py

Lines changed: 9 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,18 +1,17 @@
1-
from logging import getLogger
2-
3-
logger = getLogger(__name__)
4-
logger.setLevel("INFO")
5-
1+
from sklearn.utils.estimator_checks import parametrize_with_checks
62
def test_import():
73
import radius_clustering as rad
84

95

106
def test_from_import():
117
from radius_clustering import RadiusClustering
128

13-
def test_check_estimator_api_consistency():
14-
from radius_clustering import RadiusClustering
15-
from sklearn.utils.estimator_checks import check_estimator
169

17-
# Check the API consistency of the RadiusClustering estimator
18-
check_estimator(RadiusClustering())
10+
from radius_clustering import RadiusClustering
11+
12+
@parametrize_with_checks([RadiusClustering()])
13+
def test_check_estimator_api_consistency(estimator, check, request):
14+
15+
"""Check the API consistency of the RadiusClustering estimator
16+
"""
17+
check(estimator)

0 commit comments

Comments
 (0)