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

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