package de.ugoe.cs.quest.usageprofiles;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Random;
import org.junit.*;
import de.ugoe.cs.quest.eventcore.Event;
import de.ugoe.cs.quest.eventcore.StringEventType;
import de.ugoe.cs.quest.usageprofiles.FirstOrderMarkovModel;
import de.ugoe.cs.quest.usageprofiles.FirstOrderMarkovModel.MarkovEdge;
import static org.junit.Assert.*;
/**
* The class FirstOrderMarkovModelTest
contains tests for the class
* {@link FirstOrderMarkovModel}
.
*
* @author Steffen Herbold
* @version 1.0
*/
public class FirstOrderMarkovModelTest {
Collection> sequences;
@Test
public void testFirstOrderMarkovModel_1() throws Exception {
Random r = new Random();
FirstOrderMarkovModel result = new FirstOrderMarkovModel(r);
assertNotNull(result);
assertEquals(r, result.r);
assertEquals(2, result.trieOrder);
}
@Test(expected = java.security.InvalidParameterException.class)
public void testFirstOrderMarkovModel_2() throws Exception {
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);
}
@Test
public void testMarkovEdgeMarkovEdge_1() throws Exception {
double weight = 0.2d;
MarkovEdge result = new MarkovEdge(weight);
assertNotNull(result);
assertEquals(weight, result.weight, 0.0001);
}
@Test
public void testMarkovEdgeToString_1() throws Exception {
double weight = 0.2d;
MarkovEdge fixture = new MarkovEdge(weight);
String result = fixture.toString();
assertEquals(Double.toString(0.2d), result);
}
@Before
public void setUp() throws Exception {
List sequence = new ArrayList();
sequence.add(new Event(new StringEventType("a")));
sequence.add(new Event(new StringEventType("b")));
sequence.add(new Event(new StringEventType("r")));
sequence.add(new Event(new StringEventType("a")));
sequence.add(new Event(new StringEventType("c")));
sequence.add(new Event(new StringEventType("a")));
sequence.add(new Event(new StringEventType("d")));
sequence.add(new Event(new StringEventType("a")));
sequence.add(new Event(new StringEventType("b")));
sequence.add(new Event(new StringEventType("r")));
sequence.add(new Event(new StringEventType("a")));
sequences = new ArrayList>();
sequences.add(sequence);
}
public static void main(String[] args) {
new org.junit.runner.JUnitCore().run(FirstOrderMarkovModelTest.class);
}
}