From f2bafa715ae25cf5b0a1821a434d1213651698e7 Mon Sep 17 00:00:00 2001
From: Tiago Freitas Pereira <tiagofrepereira@gmail.com>
Date: Tue, 13 Jan 2015 13:23:06 +0100
Subject: [PATCH] Fixed unit test
---
bob/learn/misc/test_kmeans.py | 8 +++-----
1 file changed, 3 insertions(+), 5 deletions(-)
diff --git a/bob/learn/misc/test_kmeans.py b/bob/learn/misc/test_kmeans.py
index 26cca26..0c6cc25 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()
--
GitLab