@@ -55,23 +55,23 @@ def test_pammedsil(self):
5555 dist = np .array ([[0 , 2 , 3 , 4 , 5 ], [2 , 0 , 6 , 7 , 8 ], [3 , 6 , 0 , 9 , 10 ], [4 , 7 , 9 , 0 , 11 ], [5 , 8 , 10 , 11 , 0 ]], dtype = np .float32 )
5656 pms = kmedoids .pammedsil (dist , 2 )
5757 pms_rust = kmedoids .kmedoids ._pammedsil_swap_f32 (dist , pms .medoids , 100 )
58- assert pms .loss == 0.8172727272727272
58+ assert pms .loss == 0.8172727272727273
5959 assert pms .loss == pms_rust [0 ]
6060 assert np .array_equal (pms .medoids , pms_rust [2 ])
6161
6262 def test_fastmsc (self ):
6363 dist = np .array ([[0 , 2 , 3 , 4 , 5 ], [2 , 0 , 6 , 7 , 8 ], [3 , 6 , 0 , 9 , 10 ], [4 , 7 , 9 , 0 , 11 ], [5 , 8 , 10 , 11 , 0 ]], dtype = np .float32 )
6464 fmsc = kmedoids .fastmsc (dist , 2 , init = "build" )
6565 fmsc_rust = kmedoids .kmedoids ._fastmsc_f32 (dist , fmsc .medoids , 100 )
66- assert fmsc .loss == 0.8172727272727272
66+ assert fmsc .loss == 0.8172727272727273
6767 assert np .array_equal (fmsc .medoids , fmsc_rust [2 ])
6868 assert fmsc .loss == fmsc_rust [0 ]
6969
7070 def test_fastermsc (self ):
7171 dist = np .array ([[0 , 2 , 3 , 4 , 5 ], [2 , 0 , 6 , 7 , 8 ], [3 , 6 , 0 , 9 , 10 ], [4 , 7 , 9 , 0 , 11 ], [5 , 8 , 10 , 11 , 0 ]], dtype = np .float32 )
7272 fmsc = kmedoids .fastermsc (dist , 2 , init = "build" )
7373 fmsc_rust = kmedoids .kmedoids ._fastermsc_f32 (dist , fmsc .medoids , 100 )
74- assert fmsc .loss == 0.8172727272727272
74+ assert fmsc .loss == 0.8172727272727273
7575 assert np .array_equal (fmsc .medoids , fmsc_rust [2 ])
7676 assert fmsc .loss == fmsc_rust [0 ]
7777
@@ -81,7 +81,7 @@ def test_silhouette(self):
8181 sil = kmedoids .silhouette (dist , pam .labels , n_cpu = 1 )
8282 par_sil = kmedoids .silhouette (dist , pam .labels , n_cpu = 2 )
8383 sil_rust = kmedoids .kmedoids ._silhouette_i32 (dist , pam .labels , False )
84- assert sil [0 ] == par_sil
84+ assert sil [0 ] == par_sil [ 0 ]
8585 assert sil [0 ] == sil_rust [0 ]
8686
8787 def test_medoid_silhouette (self ):
0 commit comments