Changing things all the day. Badly more to come to prepare the lib for a demonstration in a magazine. :)

git-svn-id: svn://svn.code.sf.net/p/wiigee/code/trunk@91 c7eff9ee-dd40-0410-8832-91a4d88773cf
2009-06-09 13:06:09 +00:00
parent d6a6705bef
commit d836f00c91
6 changed files with 251 additions and 58 deletions
--- a/test/org/wiigee/logic/HMMTest.java
+++ b/test/org/wiigee/logic/HMMTest.java
@@ -0,0 +1,73 @@
+/*
+ * To change this template, choose Tools | Templates
+ * and open the template in the editor.
+ */
+
+package org.wiigee.logic;
+
+import java.util.Vector;
+import junit.framework.Assert;
+import junit.framework.TestCase;
+
+/**
+ *
+ * @author bepo
+ */
+public class HMMTest extends TestCase {
+    
+    public HMMTest(String testName) {
+        super(testName);
+    }
+
+    @Override
+    protected void setUp() throws Exception {
+        super.setUp();
+    }
+
+    @Override
+    protected void tearDown() throws Exception {
+        super.tearDown();
+    }
+
+    /**
+     * Probabilistic test of the train method, of class HMM.
+     */
+    public void testTrain() {
+        System.out.println("train");
+
+        HMM instance = new HMM(8, 14);
+
+        // setup a trainsequence
+        Vector<int[]> trainsequence = new Vector<int[]>();
+        trainsequence.add(new int[] { 0, 1, 2, 1, 6, 4, 3, 2});
+        trainsequence.add(new int[] { 0, 1, 2, 2, 6, 3, 3, 1});
+        trainsequence.add(new int[] { 0, 1, 2, 2, 6, 3, 2, 2});
+
+        // setup a failing trainsequence
+        int[] fail = new int[] { 5, 5, 5, 5, 5, 5, 5, 5};
+
+        // train the hmm
+        instance.train(trainsequence);
+
+        double probA = instance.getProbability(trainsequence.elementAt(0));
+        double probB = instance.getProbability(trainsequence.elementAt(1));
+        double probC = instance.getProbability(trainsequence.elementAt(2));
+        double probFAIL = instance.getProbability(fail);
+
+        System.out.println("probA = "+probA);
+        System.out.println("probB = "+probB);
+        System.out.println("probC = "+probC);
+        System.out.println("probFAIL = "+probFAIL);
+
+        if((probA <= 1.0E-10) ||
+           (probB <= 1.0E-10) ||
+           (probC <= 1.0E-10)) {
+            fail("Probabilities to low to be accurate.");
+        }
+
+        if(probFAIL > 0.0) {
+            fail("Fake probability to high.");
+        }
+    }
+
+}