@@ -41,19 +41,22 @@ def exact_results():
41
41
}
42
42
return results
43
43
44
- def assert_results (results , expected ):
44
+ def assert_results_exact (results , expected ):
45
45
"""Helper function to assert clustering results."""
46
- assert len (results . labels_ ) == len ( expected [ 'labels' ]), "Labels length mismatch"
46
+ assert_results (results , expected )
47
47
assert set (results .labels_ ) == set (expected ['labels' ]), "Labels do not match expected"
48
48
assert results .centers_ == expected ['centers' ], "Centers do not match expected"
49
- assert abs (results .mds_exec_time_ - expected ['time' ]) < 0.1 , "Execution time mismatch by more than 0.1 seconds"
50
- assert abs (results .effective_radius_ - expected ['effective_radius' ]) < 0.01 , "Effective radius mismatch"
51
49
assert np .sum (results .labels_ - expected ['labels' ]) == 0 , "Labels do not match expected"
52
50
51
+ def assert_results (results , expected ):
52
+ assert len (results .labels_ ) == len (expected ['labels' ]), "Labels length mismatch"
53
+ assert abs (results .mds_exec_time_ - expected ['time' ]) < 0.1 , "Execution time mismatch by more than 10%"
54
+ assert abs (results .effective_radius_ - expected ['effective_radius' ])/ results .effective_radius_ < 0.1 , "Effective radius mismatch"
55
+
53
56
def test_exact (iris_data , exact_results ):
54
57
"""Test the RadiusClustering with exact"""
55
58
clustering = RadiusClustering (radius = 1.43 , manner = 'exact' ).fit (iris_data )
56
- assert_results (clustering , exact_results )
59
+ assert_results_exact (clustering , exact_results )
57
60
58
61
def test_approx (iris_data , approx_results ):
59
62
"""Test the RadiusClustering with approx."""
0 commit comments