diff --git a/bob/learn/misc/test_kmeans.py b/bob/learn/misc/test_kmeans.py index 26cca2647602e7ccccf4de5aa66726c69f851198..0c6cc253c03df95771723d0d3a925358d5b9ec8b 100644 --- a/bob/learn/misc/test_kmeans.py +++ b/bob/learn/misc/test_kmeans.py @@ -89,10 +89,8 @@ def test_KMeansMachine2(): variances_result = numpy.array([[ 0.01,1.], [ 0.01555556, 0.00888889]]) - - weights_result = numpy.array([ 0.4, 0.6]) - - assert equals(weights_result,weights, 1e-3) - assert equals(variances_result,variances,1e-3) + + assert equals(weights_result,weights, 1e-3).all() + assert equals(variances_result,variances,1e-3).all()