Fixed Coreset test bug

Author: PyDataBlog

test/test07_coreset.jl

@@ -54,9 +54,23 @@ end
     res = kmeans(Coreset(), X, 2; tol = 1e-16, metric=Cityblock(), rng = rng)
 
     @test res.assignments == [2, 2, 1]
-    @test res.centers == [4.0 1.4865168535972686]
+    @test res.centers ≈ [4.0 1.4865168535972686]
     @test res.totalcost == 1.0
     @test res.converged
+
+
+    rng = StableRNG(2020)
+    X = rand(rng, 3, 100)
+    rng_orig = deepcopy(rng)
+
+    baseline = kmeans(Lloyd(), X, 10, tol = 1e-10, metric=Cityblock(), rng = rng, n_threads = 1)
+    rng = deepcopy(rng_orig)
+    res = kmeans(Coreset(), X, 10; tol = 1e-10, metric = Cityblock(), rng = rng, n_threads = 1)
+
+    @test res.converged == baseline.converged
+    @test res.iterations == baseline.iterations
+    @test round(res.totalcost) == round(baseline.totalcost)
+
 end
 
 end # module