Changeset 344 for trunk/EventBenchCoreTest/src

Timestamp:

12/22/11 10:13:18 (13 years ago)

Author:

sherbold

Message:

• added JUnit test case for de.ugoe.cs.eventbench.models.FirstOrderMarkovModel?.calcEntropy()

File:

• trunk/EventBenchCoreTest/src/de/ugoe/cs/eventbench/models/FirstOrderMarkovModelTest.java (modified) (3 diffs)

trunk/EventBenchCoreTest/src/de/ugoe/cs/eventbench/models/FirstOrderMarkovModelTest.java

@@ -1 +1 @@
 package de.ugoe.cs.eventbench.models;
 
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.List;
 import java.util.Random;
 import org.junit.*;
+
+import de.ugoe.cs.eventbench.data.Event;
 import static org.junit.Assert.*;
 
@@ -14 +19 @@
 public class FirstOrderMarkovModelTest {
 
+        Collection<List<? extends Event<?>>> sequences;
+        
         @Test
         public void testFirstOrderMarkovModel_1() throws Exception {
@@ -29 +36 @@
                 new FirstOrderMarkovModel(null);
         }
+        
+        @Test
+        public void testCalcEntropy() throws Exception {
+                Random r = new Random();
+                FirstOrderMarkovModel fixture = new FirstOrderMarkovModel(r);
+                fixture.train(sequences);
+                
+                double result = fixture.calcEntropy();
+                
+                assertEquals(0.7392d, result, 0.0001);
+        }
+        
+        @Before
+        public void setUp() throws Exception {
+                List<Event<?>> sequence = new ArrayList<Event<?>>();
+                sequence.add(new Event<String>("a"));
+                sequence.add(new Event<String>("b"));
+                sequence.add(new Event<String>("r"));
+                sequence.add(new Event<String>("a"));
+                sequence.add(new Event<String>("c"));
+                sequence.add(new Event<String>("a"));
+                sequence.add(new Event<String>("d"));
+                sequence.add(new Event<String>("a"));
+                sequence.add(new Event<String>("b"));
+                sequence.add(new Event<String>("r"));
+                sequence.add(new Event<String>("a"));
+
+                sequences = new ArrayList<List<? extends Event<?>>>();
+                sequences.add(sequence);
+        }
 
         public static void main(String[] args) {