src/test/java/jp/sourceforge/stigmata/birthmarks/BirthmarkCompareTest.java

   1 package jp.sourceforge.stigmata.birthmarks;
   2
   3 /*
   4  * $Id$
   5  */
   6
   7 import jp.sourceforge.stigmata.BirthmarkContext;
   8 import jp.sourceforge.stigmata.BirthmarkEngine;
   9 import jp.sourceforge.stigmata.BirthmarkSet;
  10 import jp.sourceforge.stigmata.ExtractionResultSet;
  11 import jp.sourceforge.stigmata.ExtractionTarget;
  12 import jp.sourceforge.stigmata.Stigmata;
  13
  14 import org.junit.Assert;
  15 import org.junit.Before;
  16 import org.junit.Test;
  17
  18 /**
  19  *
  20  * @author Haruaki Tamada
  21  */
  22 public class BirthmarkCompareTest{
  23     private BirthmarkContext context;
  24     private BirthmarkSet[] birthmarks;
  25     private BirthmarkEngine engine;
  26
  27     @Before
  28     public void readBirthmarks() throws Exception{
  29         context = Stigmata.getInstance().createContext();
  30         context.setBirthmarkTypes(new String[] { "cvfv", "smc", "uc", "is", "kgram", });
  31         engine = new BirthmarkEngine(context.getEnvironment());
  32
  33         ExtractionResultSet ers = engine.extract(
  34             new String[] {
  35                 "target/classes/jp/sourceforge/stigmata/Stigmata.class",
  36                 "target/classes/jp/sourceforge/stigmata/Main.class",
  37                 "target/classes/jp/sourceforge/stigmata/BirthmarkEnvironment.class",
  38             }, context
  39         );
  40         birthmarks = ers.getBirthmarkSets(ExtractionTarget.TARGET_BOTH);
  41     }
  42
  43     @Test
  44     public void equalsTest() throws Exception{
  45         Assert.assertEquals(3, birthmarks.length);
  46         BirthmarkEngine engine = Stigmata.getInstance().createEngine();
  47
  48         Assert.assertEquals(1d, engine.compareDetails(birthmarks[0], birthmarks[0], context).calculateSimilarity(), 1E-6);
  49         Assert.assertEquals(1d, engine.compareDetails(birthmarks[1], birthmarks[1], context).calculateSimilarity(), 1E-6);
  50         Assert.assertEquals(1d, engine.compareDetails(birthmarks[2], birthmarks[2], context).calculateSimilarity(), 1E-6);
  51     }
  52 }