From 694357028cab929d7ffa871b993903a69076de09 Mon Sep 17 00:00:00 2001
From: Tiago Freitas Pereira <tiagofrepereira@gmail.com>
Date: Mon, 12 Jan 2015 15:10:11 +0100
Subject: [PATCH] Added more unit tests
---
bob/learn/misc/test_kmeans.py | 13 +++++--------
1 file changed, 5 insertions(+), 8 deletions(-)
diff --git a/bob/learn/misc/test_kmeans.py b/bob/learn/misc/test_kmeans.py
index 69c3e6a..26cca26 100644
--- a/bob/learn/misc/test_kmeans.py
+++ b/bob/learn/misc/test_kmeans.py
@@ -75,7 +75,6 @@ def test_KMeansMachine():
def test_KMeansMachine2():
- print "Computing"
kmeans = bob.learn.misc.KMeansMachine(2,2)
kmeans.means = numpy.array([[1.2,1.3],[0.2,-0.3]])
@@ -86,16 +85,14 @@ def test_KMeansMachine2():
[0.3,0.2],
[0.2,0]
])
- print "Computing"
variances, weights = kmeans.get_variances_and_weights_for_each_cluster(data)
- print "Computed"
variances_result = numpy.array([[ 0.01,1.],
- [ 0.01555556 ,0.00888889]])
+ [ 0.01555556, 0.00888889]])
+
+
weights_result = numpy.array([ 0.4, 0.6])
- assert True
-
- #assert weights_result == weights
- #assert variances_result == variances
+ assert equals(weights_result,weights, 1e-3)
+ assert equals(variances_result,variances,1e-3)
--
GitLab