src/test/java/jp/naist/se/stigmata/birthmarks/BirthmarkCompareTest.java

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