1 package jp.sourceforge.stigmata.birthmarks.kgram;
7 import org.junit.Assert;
8 import org.junit.Before;
13 * @author Haruaki Tamada
15 public class KGramTest{
16 private String[] plainValues;
17 private String[] complexValues;
20 public void buildKGrams(){
21 plainValues = new String[] { "a", "b", "c", "d", "e", "f", "g", };
22 complexValues = new String[] { "a", "b", "r", "a", "c", "a", "d", "a", "b", "r", "a", };
26 public void checkPlainKGram(){
27 KGram<String>[] kgrams = KGramBuilder.getInstance().buildKGram(plainValues, 4);
28 Assert.assertEquals(4, kgrams.length);
30 Assert.assertEquals(4, kgrams[0].getKValue());
31 Assert.assertEquals(4, kgrams[1].getKValue());
32 Assert.assertEquals(4, kgrams[2].getKValue());
33 Assert.assertEquals(4, kgrams[3].getKValue());
35 Assert.assertArrayEquals(new String[] { "a", "b", "c", "d", }, kgrams[0].toArray());
36 Assert.assertArrayEquals(new String[] { "b", "c", "d", "e", }, kgrams[1].toArray());
37 Assert.assertArrayEquals(new String[] { "c", "d", "e", "f", }, kgrams[2].toArray());
38 Assert.assertArrayEquals(new String[] { "d", "e", "f", "g", }, kgrams[3].toArray());
42 public void checkPlainKGram2(){
43 KGram<String>[] kgrams = KGramBuilder.getInstance().buildKGram(plainValues, 3);
44 Assert.assertEquals(5, kgrams.length);
46 Assert.assertEquals(3, kgrams[0].getKValue());
47 Assert.assertEquals(3, kgrams[1].getKValue());
48 Assert.assertEquals(3, kgrams[2].getKValue());
49 Assert.assertEquals(3, kgrams[3].getKValue());
50 Assert.assertEquals(3, kgrams[4].getKValue());
52 Assert.assertArrayEquals(new String[] { "a", "b", "c", }, kgrams[0].toArray());
53 Assert.assertArrayEquals(new String[] { "b", "c", "d", }, kgrams[1].toArray());
54 Assert.assertArrayEquals(new String[] { "c", "d", "e", }, kgrams[2].toArray());
55 Assert.assertArrayEquals(new String[] { "d", "e", "f", }, kgrams[3].toArray());
56 Assert.assertArrayEquals(new String[] { "e", "f", "g", }, kgrams[4].toArray());
60 public void checkComplexKGram(){
61 KGram<String>[] kgrams = KGramBuilder.getInstance().buildKGram(complexValues, 3);
63 Assert.assertEquals(7, kgrams.length);
65 Assert.assertArrayEquals(new String[] { "a", "b", "r", }, kgrams[0].toArray());
66 Assert.assertArrayEquals(new String[] { "b", "r", "a", }, kgrams[1].toArray());
67 Assert.assertArrayEquals(new String[] { "r", "a", "c", }, kgrams[2].toArray());
68 Assert.assertArrayEquals(new String[] { "a", "c", "a", }, kgrams[3].toArray());
69 Assert.assertArrayEquals(new String[] { "c", "a", "d", }, kgrams[4].toArray());
70 Assert.assertArrayEquals(new String[] { "a", "d", "a", }, kgrams[5].toArray());
71 Assert.assertArrayEquals(new String[] { "d", "a", "b", }, kgrams[6].toArray());
72 // following kgram is appeared in above.
73 // assertEquals(new String[] { "a", "b", "r", }, kgrams[0].toArray());
74 // assertEquals(new String[] { "b", "r", "a", }, kgrams[1].toArray());