src/test/java/jp/sourceforge/stigmata/birthmarks/kgram/KGramTest.java

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