Context Navigation

TrieBasedModelTest.java @ 344

Last change on this file since 344 was 343, checked in by sherbold, 13 years ago
*added JUnit test cases for de.ugoe.cs.eventbench.models.DeterministicFiniteAutomaton?, HighOrderMarkovModel?, FirstOrderMarkovModel?, and PredictionByPartialMatch?
Property svn:mime-type set to `text/plain`
File size: 21.2 KB

Rev	Line
[257]	1	package de.ugoe.cs.eventbench.models;
	2
	3	import java.util.ArrayList;
	4	import java.util.Collection;
	5	import java.util.HashSet;
	6	import java.util.List;
	7	import java.util.Random;
	8	import de.ugoe.cs.eventbench.data.Event;
	9	import org.junit.*;
	10	import static org.junit.Assert.*;
	11
	12	/**
[343]	13	* The class <code>TrieBasedModelTest</code> contains tests for the class
	14	* <code>{@link TrieBasedModel}</code>.
	15	*
[257]	16	* @author Steffen Herbold
	17	* @version 1.0
	18	*/
	19	public class TrieBasedModelTest {
[343]	20
[257]	21	List<Event<?>> sequence;
	22	Collection<Event<?>> symbols;
[343]	23
[257]	24	private void assertTrieStructure(Trie<Event<?>> trie, int numSequences) {
	25	TrieNode<Event<?>> root = trie.find(null);
	26	TrieNode<Event<?>> root_a = root.getChild(new Event<String>("a"));
	27	TrieNode<Event<?>> root_a_a = root_a.getChild(new Event<String>("a"));
	28	TrieNode<Event<?>> root_a_b = root_a.getChild(new Event<String>("b"));
[343]	29	TrieNode<Event<?>> root_a_b_a = root_a_b
	30	.getChild(new Event<String>("a"));
	31	TrieNode<Event<?>> root_a_b_b = root_a_b
	32	.getChild(new Event<String>("b"));
	33	TrieNode<Event<?>> root_a_b_c = root_a_b
	34	.getChild(new Event<String>("c"));
	35	TrieNode<Event<?>> root_a_b_d = root_a_b
	36	.getChild(new Event<String>("d"));
	37	TrieNode<Event<?>> root_a_b_r = root_a_b
	38	.getChild(new Event<String>("r"));
[257]	39	TrieNode<Event<?>> root_a_c = root_a.getChild(new Event<String>("c"));
[343]	40	TrieNode<Event<?>> root_a_c_a = root_a_c
	41	.getChild(new Event<String>("a"));
	42	TrieNode<Event<?>> root_a_c_b = root_a_c
	43	.getChild(new Event<String>("b"));
	44	TrieNode<Event<?>> root_a_c_c = root_a_c
	45	.getChild(new Event<String>("c"));
	46	TrieNode<Event<?>> root_a_c_d = root_a_c
	47	.getChild(new Event<String>("d"));
	48	TrieNode<Event<?>> root_a_c_r = root_a_c
	49	.getChild(new Event<String>("r"));
[257]	50	TrieNode<Event<?>> root_a_d = root_a.getChild(new Event<String>("d"));
[343]	51	TrieNode<Event<?>> root_a_d_a = root_a_d
	52	.getChild(new Event<String>("a"));
	53	TrieNode<Event<?>> root_a_d_b = root_a_d
	54	.getChild(new Event<String>("b"));
	55	TrieNode<Event<?>> root_a_d_c = root_a_d
	56	.getChild(new Event<String>("c"));
	57	TrieNode<Event<?>> root_a_d_d = root_a_d
	58	.getChild(new Event<String>("d"));
	59	TrieNode<Event<?>> root_a_d_r = root_a_d
	60	.getChild(new Event<String>("r"));
[257]	61	TrieNode<Event<?>> root_a_r = root_a.getChild(new Event<String>("r"));
	62	TrieNode<Event<?>> root_b = root.getChild(new Event<String>("b"));
	63	TrieNode<Event<?>> root_b_a = root_b.getChild(new Event<String>("a"));
	64	TrieNode<Event<?>> root_b_b = root_b.getChild(new Event<String>("b"));
	65	TrieNode<Event<?>> root_b_c = root_b.getChild(new Event<String>("c"));
	66	TrieNode<Event<?>> root_b_d = root_b.getChild(new Event<String>("d"));
	67	TrieNode<Event<?>> root_b_r = root_b.getChild(new Event<String>("r"));
[343]	68	TrieNode<Event<?>> root_b_r_a = root_b_r
	69	.getChild(new Event<String>("a"));
	70	TrieNode<Event<?>> root_b_r_b = root_b_r
	71	.getChild(new Event<String>("b"));
	72	TrieNode<Event<?>> root_b_r_c = root_b_r
	73	.getChild(new Event<String>("c"));
	74	TrieNode<Event<?>> root_b_r_d = root_b_r
	75	.getChild(new Event<String>("d"));
	76	TrieNode<Event<?>> root_b_r_r = root_b_r
	77	.getChild(new Event<String>("r"));
[257]	78	TrieNode<Event<?>> root_c = root.getChild(new Event<String>("c"));
	79	TrieNode<Event<?>> root_c_a = root_c.getChild(new Event<String>("a"));
[343]	80	TrieNode<Event<?>> root_c_a_a = root_c_a
	81	.getChild(new Event<String>("a"));
	82	TrieNode<Event<?>> root_c_a_b = root_c_a
	83	.getChild(new Event<String>("b"));
	84	TrieNode<Event<?>> root_c_a_c = root_c_a
	85	.getChild(new Event<String>("c"));
	86	TrieNode<Event<?>> root_c_a_d = root_c_a
	87	.getChild(new Event<String>("d"));
	88	TrieNode<Event<?>> root_c_a_r = root_c_a
	89	.getChild(new Event<String>("r"));
[257]	90	TrieNode<Event<?>> root_c_b = root_c.getChild(new Event<String>("b"));
	91	TrieNode<Event<?>> root_c_c = root_c.getChild(new Event<String>("c"));
	92	TrieNode<Event<?>> root_c_d = root_c.getChild(new Event<String>("d"));
	93	TrieNode<Event<?>> root_c_r = root_c.getChild(new Event<String>("r"));
	94	TrieNode<Event<?>> root_d = root.getChild(new Event<String>("d"));
	95	TrieNode<Event<?>> root_d_a = root_d.getChild(new Event<String>("a"));
[343]	96	TrieNode<Event<?>> root_d_a_a = root_d_a
	97	.getChild(new Event<String>("a"));
	98	TrieNode<Event<?>> root_d_a_b = root_d_a
	99	.getChild(new Event<String>("b"));
	100	TrieNode<Event<?>> root_d_a_c = root_d_a
	101	.getChild(new Event<String>("c"));
	102	TrieNode<Event<?>> root_d_a_d = root_d_a
	103	.getChild(new Event<String>("d"));
	104	TrieNode<Event<?>> root_d_a_r = root_d_a
	105	.getChild(new Event<String>("r"));
[257]	106	TrieNode<Event<?>> root_d_b = root_d.getChild(new Event<String>("b"));
	107	TrieNode<Event<?>> root_d_c = root_d.getChild(new Event<String>("c"));
	108	TrieNode<Event<?>> root_d_d = root_d.getChild(new Event<String>("d"));
	109	TrieNode<Event<?>> root_d_r = root_d.getChild(new Event<String>("r"));
	110	TrieNode<Event<?>> root_r = root.getChild(new Event<String>("r"));
	111	TrieNode<Event<?>> root_r_a = root_r.getChild(new Event<String>("a"));
[343]	112	TrieNode<Event<?>> root_r_a_a = root_r_a
	113	.getChild(new Event<String>("a"));
	114	TrieNode<Event<?>> root_r_a_b = root_r_a
	115	.getChild(new Event<String>("b"));
	116	TrieNode<Event<?>> root_r_a_c = root_r_a
	117	.getChild(new Event<String>("c"));
	118	TrieNode<Event<?>> root_r_a_d = root_r_a
	119	.getChild(new Event<String>("d"));
	120	TrieNode<Event<?>> root_r_a_r = root_r_a
	121	.getChild(new Event<String>("r"));
[257]	122	TrieNode<Event<?>> root_r_a_end = root_r_a.getChild(Event.ENDEVENT);
	123	TrieNode<Event<?>> root_r_b = root_r.getChild(new Event<String>("b"));
	124	TrieNode<Event<?>> root_r_c = root_r.getChild(new Event<String>("c"));
	125	TrieNode<Event<?>> root_r_d = root_r.getChild(new Event<String>("d"));
	126	TrieNode<Event<?>> root_r_r = root_r.getChild(new Event<String>("r"));
	127	TrieNode<Event<?>> root_start = root.getChild(Event.STARTEVENT);
[343]	128	TrieNode<Event<?>> root_start_a = root_start
	129	.getChild(new Event<String>("a"));
	130	TrieNode<Event<?>> root_start_a_a = root_start_a
	131	.getChild(new Event<String>("a"));
	132	TrieNode<Event<?>> root_start_a_b = root_start_a
	133	.getChild(new Event<String>("b"));
	134	TrieNode<Event<?>> root_start_a_c = root_start_a
	135	.getChild(new Event<String>("c"));
	136	TrieNode<Event<?>> root_start_a_d = root_start_a
	137	.getChild(new Event<String>("d"));
	138	TrieNode<Event<?>> root_start_a_r = root_start_a
	139	.getChild(new Event<String>("r"));
	140	TrieNode<Event<?>> root_start_b = root_start
	141	.getChild(new Event<String>("b"));
	142	TrieNode<Event<?>> root_start_c = root_start
	143	.getChild(new Event<String>("c"));
	144	TrieNode<Event<?>> root_start_d = root_start
	145	.getChild(new Event<String>("d"));
	146	TrieNode<Event<?>> root_start_r = root_start
	147	.getChild(new Event<String>("r"));
	148
	149	assertEquals(numSequences * 5, root_a.getCount());
[257]	150	assertNull(root_a_a);
[343]	151	assertEquals(numSequences * 2, root_a_b.getCount());
[257]	152	assertNull(root_a_b_a);
	153	assertNull(root_a_b_b);
	154	assertNull(root_a_b_c);
	155	assertNull(root_a_b_d);
[343]	156	assertEquals(numSequences * 2, root_a_b_r.getCount());
	157	assertEquals(numSequences * 1, root_a_c.getCount());
	158	assertEquals(numSequences * 1, root_a_c_a.getCount());
[257]	159	assertNull(root_a_c_b);
	160	assertNull(root_a_c_c);
	161	assertNull(root_a_c_d);
	162	assertNull(root_a_c_r);
[343]	163	assertEquals(numSequences * 1, root_a_d.getCount());
	164	assertEquals(numSequences * 1, root_a_d_a.getCount());
[257]	165	assertNull(root_a_d_b);
	166	assertNull(root_a_d_c);
	167	assertNull(root_a_d_d);
	168	assertNull(root_a_d_r);
	169	assertNull(root_a_r);
[343]	170
	171	assertEquals(numSequences * 2, root_b.getCount());
[257]	172	assertNull(root_b_a);
	173	assertNull(root_b_b);
	174	assertNull(root_b_c);
	175	assertNull(root_b_d);
[343]	176	assertEquals(numSequences * 2, root_b_r.getCount());
	177	assertEquals(numSequences * 2, root_b_r_a.getCount());
[257]	178	assertNull(root_b_r_b);
	179	assertNull(root_b_r_c);
	180	assertNull(root_b_r_d);
	181	assertNull(root_b_r_r);
[343]	182
	183	assertEquals(numSequences * 1, root_c.getCount());
	184	assertEquals(numSequences * 1, root_c_a.getCount());
[257]	185	assertNull(root_c_a_a);
	186	assertNull(root_c_a_b);
	187	assertNull(root_c_a_c);
[343]	188	assertEquals(numSequences * 1, root_c_a_d.getCount());
[257]	189	assertNull(root_c_a_r);
	190	assertNull(root_c_b);
	191	assertNull(root_c_c);
	192	assertNull(root_c_d);
	193	assertNull(root_c_r);
[343]	194
	195	assertEquals(numSequences * 1, root_d.getCount());
	196	assertEquals(numSequences * 1, root_d_a.getCount());
[257]	197	assertNull(root_d_a_a);
[343]	198	assertEquals(numSequences * 1, root_d_a_b.getCount());
[257]	199	assertNull(root_d_a_c);
	200	assertNull(root_d_a_d);
	201	assertNull(root_d_a_r);
	202	assertNull(root_d_b);
	203	assertNull(root_d_c);
	204	assertNull(root_d_d);
	205	assertNull(root_d_r);
[343]	206
	207	assertEquals(numSequences * 2, root_r.getCount());
	208	assertEquals(numSequences * 2, root_r_a.getCount());
[257]	209	assertNull(root_r_a_a);
	210	assertNull(root_r_a_b);
[343]	211	assertEquals(numSequences * 1, root_r_a_c.getCount());
[257]	212	assertNull(root_r_a_d);
	213	assertNull(root_r_a_r);
[343]	214	assertEquals(numSequences * 1, root_r_a_end.getCount());
[257]	215	assertNull(root_r_b);
	216	assertNull(root_r_c);
	217	assertNull(root_r_d);
	218	assertNull(root_r_r);
[343]	219
	220	assertEquals(numSequences * 1, root_start.getCount());
	221	assertEquals(numSequences * 1, root_start_a.getCount());
[257]	222	assertNull(root_start_a_a);
[343]	223	assertEquals(numSequences * 1, root_start_a_b.getCount());
[257]	224	assertNull(root_start_a_c);
	225	assertNull(root_start_a_d);
	226	assertNull(root_start_a_r);
	227	assertNull(root_start_b);
	228	assertNull(root_start_c);
	229	assertNull(root_start_d);
	230	assertNull(root_start_r);
[343]	231
[257]	232	// check if leafs are really leafs
	233	assertTrue(root_a_b_r.isLeaf());
	234	assertTrue(root_a_c_a.isLeaf());
	235	assertTrue(root_a_d_a.isLeaf());
	236	assertTrue(root_b_r_a.isLeaf());
	237	assertTrue(root_c_a_d.isLeaf());
	238	assertTrue(root_d_a_b.isLeaf());
	239	assertTrue(root_r_a_c.isLeaf());
	240	assertTrue(root_r_a_end.isLeaf());
	241	}
[343]	242
	243	private static void assertCollectionContent(Collection<?> c1,
	244	Collection<?> c2) {
[257]	245	assertEquals(c1.size(), c2.size());
[343]	246	for (Object obj : c1) {
[257]	247	assertTrue(c2.contains(obj));
	248	}
	249	}
[343]	250
[257]	251	@Test
[343]	252	public void testTrieBasedModel_1() throws Exception {
[257]	253	int markovOrder = 2;
[343]	254	Random r = new Random();
	255
	256	MockTrieBasedModel result = new MockTrieBasedModel(markovOrder, r);
	257
	258	assertNotNull(result);
	259	assertEquals(markovOrder + 1, result.trieOrder);
	260	assertEquals(r, result.r);
	261	}
	262
	263	@Test(expected = java.security.InvalidParameterException.class)
	264	public void testTrieBasedModel_2() throws Exception {
	265	int markovOrder = -1;
	266	Random r = new Random();
	267
	268	new MockTrieBasedModel(markovOrder, r);
	269	}
	270
	271	@Test(expected = java.security.InvalidParameterException.class)
	272	public void testTrieBasedModel_3() throws Exception {
	273	int markovOrder = 2;
	274	Random r = null;
	275
	276	new MockTrieBasedModel(markovOrder, r);
	277	}
	278
	279	@Test
	280	public void testGenerateSequences_1() throws Exception {
	281	int markovOrder = 2;
	282	MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder,
	283	new Random());
[326]	284	Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>();
[257]	285	sequences.add(sequence);
	286	fixture.train(sequences);
	287	int length = 2;
[343]	288
[257]	289	Collection<List<Event<?>>> expected = new HashSet<List<Event<?>>>();
	290	ArrayList<Event<?>> list;
	291	list = new ArrayList<Event<?>>();
	292	list.add(new Event<String>("a"));
	293	list.add(Event.ENDEVENT);
	294	expected.add(list);
	295	list = new ArrayList<Event<?>>();
	296	list.add(new Event<String>("a"));
	297	list.add(new Event<String>("b"));
	298	expected.add(list);
	299	list = new ArrayList<Event<?>>();
	300	list.add(new Event<String>("a"));
	301	list.add(new Event<String>("c"));
	302	expected.add(list);
	303	list = new ArrayList<Event<?>>();
	304	list.add(new Event<String>("a"));
	305	list.add(new Event<String>("d"));
	306	expected.add(list);
	307	list = new ArrayList<Event<?>>();
	308	list.add(new Event<String>("b"));
	309	list.add(new Event<String>("r"));
	310	expected.add(list);
	311	list = new ArrayList<Event<?>>();
	312	list.add(new Event<String>("c"));
	313	list.add(new Event<String>("a"));
	314	expected.add(list);
	315	list = new ArrayList<Event<?>>();
	316	list.add(new Event<String>("d"));
	317	list.add(new Event<String>("a"));
	318	expected.add(list);
	319	list = new ArrayList<Event<?>>();
	320	list.add(new Event<String>("r"));
	321	list.add(new Event<String>("a"));
	322	expected.add(list);
	323	list = new ArrayList<Event<?>>();
	324	list.add(Event.STARTEVENT);
	325	list.add(new Event<String>("a"));
	326	expected.add(list);
	327
[343]	328	Collection<List<? extends Event<?>>> result = fixture
	329	.generateSequences(length);
[257]	330
	331	assertCollectionContent(expected, result);
	332	}
	333
	334	@Test
[343]	335	public void testGenerateSequences_2() throws Exception {
[257]	336	int markovOrder = 2;
[343]	337	MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder,
	338	new Random());
[326]	339	Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>();
[257]	340	sequences.add(sequence);
	341	fixture.train(sequences);
	342	int length = 3;
[343]	343
[257]	344	Collection<List<Event<?>>> expected = new HashSet<List<Event<?>>>();
	345	ArrayList<Event<?>> list;
	346	list = new ArrayList<Event<?>>();
	347	list.add(Event.STARTEVENT);
	348	list.add(new Event<String>("a"));
	349	list.add(Event.ENDEVENT);
	350	expected.add(list);
	351	list = new ArrayList<Event<?>>();
	352	list.add(Event.STARTEVENT);
	353	list.add(new Event<String>("a"));
	354	list.add(new Event<String>("b"));
	355	expected.add(list);
	356	list = new ArrayList<Event<?>>();
	357	list.add(Event.STARTEVENT);
	358	list.add(new Event<String>("a"));
	359	list.add(new Event<String>("c"));
	360	expected.add(list);
	361	list = new ArrayList<Event<?>>();
	362	list.add(Event.STARTEVENT);
	363	list.add(new Event<String>("a"));
	364	list.add(new Event<String>("d"));
	365	expected.add(list);
	366
[343]	367	Collection<List<? extends Event<?>>> result = fixture
	368	.generateSequences(length, true);
[257]	369
	370	assertCollectionContent(expected, result);
	371	}
	372
[343]	373	@Test(expected = java.security.InvalidParameterException.class)
	374	public void testGenerateSequences_3() throws Exception {
[257]	375	int markovOrder = 2;
[343]	376	MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder,
	377	new Random());
[326]	378	Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>();
[257]	379	sequences.add(sequence);
	380	fixture.train(sequences);
	381	int length = 0;
	382
	383	fixture.generateSequences(length, false);
	384	}
	385
	386	@Test
[343]	387	public void testGenerateValidSequences_1() throws Exception {
[257]	388	int markovOrder = 2;
[343]	389	MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder,
	390	new Random());
[326]	391	Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>();
[257]	392	sequences.add(sequence);
	393	fixture.train(sequences);
	394	int length = 5;
[343]	395
[257]	396	Collection<List<Event<?>>> expected = new HashSet<List<Event<?>>>();
	397	ArrayList<Event<?>> list;
	398	list = new ArrayList<Event<?>>();
	399	list.add(Event.STARTEVENT);
	400	list.add(new Event<String>("a"));
	401	list.add(new Event<String>("c"));
	402	list.add(new Event<String>("a"));
	403	list.add(Event.ENDEVENT);
	404	expected.add(list);
	405	list = new ArrayList<Event<?>>();
	406	list.add(Event.STARTEVENT);
	407	list.add(new Event<String>("a"));
	408	list.add(new Event<String>("d"));
	409	list.add(new Event<String>("a"));
	410	list.add(Event.ENDEVENT);
	411	expected.add(list);
	412
[343]	413	Collection<List<? extends Event<?>>> result = fixture
	414	.generateValidSequences(length);
[257]	415
	416	assertCollectionContent(expected, result);
	417	}
	418
[343]	419	@Test(expected = java.security.InvalidParameterException.class)
	420	public void testGenerateValidSequences_2() throws Exception {
[257]	421	int markovOrder = 2;
[343]	422	MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder,
	423	new Random());
[326]	424	Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>();
[257]	425	sequences.add(sequence);
	426	fixture.train(sequences);
	427	int length = 0;
	428
	429	fixture.generateValidSequences(length);
	430	}
[343]	431
[257]	432	@Test
[343]	433	public void testGetEvents_1() throws Exception {
[257]	434	int markovOrder = 2;
[343]	435	MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder,
	436	new Random());
[326]	437	Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>();
[257]	438	sequences.add(sequence);
	439
	440	fixture.train(sequences);
	441
	442	Collection<? extends Event<?>> result = fixture.getEvents();
	443
	444	assertCollectionContent(symbols, result);
	445	}
[343]	446
[257]	447	@Test
[343]	448	public void testGetEvents_2() throws Exception {
[257]	449	int markovOrder = 2;
[343]	450	MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder,
	451	new Random());
	452
[257]	453	Collection<? extends Event<?>> result = fixture.getEvents();
	454
	455	assertCollectionContent(new HashSet<Event<?>>(), result);
	456	}
	457
	458	@Test
[343]	459	public void testGetNumFOMStates_1() throws Exception {
[257]	460	int markovOrder = 2;
[343]	461	MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder,
	462	new Random());
[326]	463	Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>();
[257]	464	sequences.add(sequence);
	465
	466	fixture.train(sequences);
	467
	468	int result = fixture.getNumFOMStates();
	469
	470	assertEquals(10, result);
	471	}
[343]	472
[257]	473	@Test
[343]	474	public void testGetNumFOMStates_2() throws Exception {
[257]	475	int markovOrder = 2;
[343]	476	MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder,
	477	new Random());
	478	;
[257]	479
	480	int result = fixture.getNumFOMStates();
	481
	482	assertEquals(0, result);
	483	}
[343]	484
[257]	485	@Test
[343]	486	public void testGetNumSymbols_1() throws Exception {
[257]	487	int markovOrder = 2;
[343]	488	MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder,
	489	new Random());
[326]	490	Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>();
[257]	491	sequences.add(sequence);
	492	fixture.train(sequences);
	493
	494	int result = fixture.getNumSymbols();
	495
	496	assertEquals(7, result);
	497	}
[343]	498
[257]	499	@Test
[343]	500	public void testGetNumSymbols_2() throws Exception {
[257]	501	int markovOrder = 2;
[343]	502	MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder,
	503	new Random());
[257]	504
	505	int result = fixture.getNumSymbols();
	506
	507	assertEquals(0, result);
	508	}
	509
	510	@Test
[343]	511	public void testGetNumTransitions_1() throws Exception {
[257]	512	int markovOrder = 2;
[343]	513	MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder,
	514	new Random());
[326]	515	Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>();
[257]	516	sequences.add(sequence);
	517	fixture.train(sequences);
[343]	518
[257]	519	int result = fixture.getNumTransitions();
	520
	521	assertEquals(11, result);
	522	}
[343]	523
[257]	524	@Test
[343]	525	public void testGetNumTransitions_2() throws Exception {
[257]	526	int markovOrder = 2;
[343]	527	MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder,
	528	new Random());
[257]	529
	530	int result = fixture.getNumTransitions();
	531
	532	assertEquals(0, result);
	533	}
	534
	535	@Test
[343]	536	public void testTrain_1() throws Exception {
[257]	537	int markovOrder = 2;
[343]	538	MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder,
	539	new Random());
[326]	540	Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>();
[257]	541	sequences.add(sequence);
	542
	543	fixture.train(sequences);
[343]	544
[257]	545	assertCollectionContent(symbols, fixture.getEvents());
[343]	546
	547	assertTrieStructure(fixture.trie, 1);
	548	}
	549
[257]	550	@Test
[343]	551	public void testTrain_2() throws Exception {
[257]	552	int markovOrder = 2;
[343]	553	MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder,
	554	new Random());
[326]	555	Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>();
[257]	556	sequences.add(sequence);
	557	sequences.add(sequence);
	558
	559	fixture.train(sequences);
[343]	560
[257]	561	assertCollectionContent(symbols, fixture.getEvents());
[343]	562
[257]	563	assertTrieStructure(fixture.trie, 2);
	564	}
	565
[343]	566	@Test(expected = java.security.InvalidParameterException.class)
	567	public void testTrain_3() throws Exception {
	568	int markovOrder = 2;
	569	MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder,
	570	new Random());
	571	Collection<List<? extends Event<?>>> sequences = null;
	572
	573	fixture.train(sequences);
	574	}
	575
[257]	576	@Test
[343]	577	public void testUpdate_1() throws Exception {
[257]	578	int markovOrder = 2;
[343]	579	MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder,
	580	new Random());
[326]	581	Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>();
[257]	582	sequences.add(sequence);
	583	fixture.train(sequences);
[343]	584
[257]	585	fixture.update(sequences);
	586
	587	assertCollectionContent(symbols, fixture.getEvents());
	588	assertTrieStructure(fixture.trie, 2);
	589	}
	590
[343]	591	@Test(expected = java.security.InvalidParameterException.class)
	592	public void testUpdate_2() throws Exception {
[257]	593	int markovOrder = 2;
[343]	594	MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder,
	595	new Random());
[326]	596	Collection<List<? extends Event<?>>> sequences = null;
[257]	597	fixture.trie = null;
	598
	599	fixture.update(sequences);
	600	}
	601
	602	@Before
[343]	603	public void setUp() throws Exception {
[257]	604	sequence = new ArrayList<Event<?>>();
	605	sequence.add(new Event<String>("a"));
	606	sequence.add(new Event<String>("b"));
	607	sequence.add(new Event<String>("r"));
	608	sequence.add(new Event<String>("a"));
	609	sequence.add(new Event<String>("c"));
	610	sequence.add(new Event<String>("a"));
	611	sequence.add(new Event<String>("d"));
	612	sequence.add(new Event<String>("a"));
	613	sequence.add(new Event<String>("b"));
	614	sequence.add(new Event<String>("r"));
	615	sequence.add(new Event<String>("a"));
[343]	616
[257]	617	symbols = new HashSet<Event<?>>();
	618	symbols.add(new Event<String>("a"));
	619	symbols.add(new Event<String>("b"));
	620	symbols.add(new Event<String>("c"));
	621	symbols.add(new Event<String>("d"));
	622	symbols.add(new Event<String>("r"));
	623	symbols.add(Event.STARTEVENT);
	624	symbols.add(Event.ENDEVENT);
	625	}
	626
	627	public static void main(String[] args) {
	628	new org.junit.runner.JUnitCore().run(TrieBasedModelTest.class);
	629	}
	630	}

Note: See TracBrowser for help on using the repository browser.

Download in other formats:

Original Format