/* | |
* Licensed to the Apache Software Foundation (ASF) under one | |
* or more contributor license agreements. See the NOTICE file | |
* distributed with this work for additional information | |
* regarding copyright ownership. The ASF licenses this file | |
* to you under the Apache License, Version 2.0 (the | |
* "License"); you may not use this file except in compliance | |
* with the License. You may obtain a copy of the License at | |
* | |
* http://www.apache.org/licenses/LICENSE-2.0 | |
* | |
* Unless required by applicable law or agreed to in writing, | |
* software distributed under the License is distributed on an | |
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY | |
* KIND, either express or implied. See the License for the | |
* specific language governing permissions and limitations | |
* under the License. | |
*/ | |
package org.apache.uima.ruta; | |
import java.util.ArrayList; | |
import java.util.HashMap; | |
import java.util.List; | |
import java.util.Map; | |
import java.util.TreeMap; | |
import org.apache.uima.cas.CAS; | |
import org.apache.uima.ruta.engine.RutaEngine; | |
import org.apache.uima.ruta.engine.RutaTestUtils; | |
import org.apache.uima.ruta.engine.RutaTestUtils.TestFeature; | |
import org.junit.Test; | |
public class FeatureMatch1Test { | |
@Test | |
public void test() throws Exception { | |
String name = this.getClass().getSimpleName(); | |
String namespace = this.getClass().getPackage().getName().replaceAll("\\.", "/"); | |
Map<String, String> complexTypes = new HashMap<String, String>(); | |
Map<String, List<TestFeature>> features = new TreeMap<String, List<TestFeature>>(); | |
String typeNameA = "org.apache.uima.ruta.FeatureMatchTest.A"; | |
String typeNameB = "org.apache.uima.ruta.FeatureMatchTest.B"; | |
String typeNameC = "org.apache.uima.ruta.FeatureMatchTest.C"; | |
String typeNameD = "org.apache.uima.ruta.FeatureMatchTest.D"; | |
complexTypes.put(typeNameA, "uima.tcas.Annotation"); | |
complexTypes.put(typeNameB, typeNameD); | |
complexTypes.put(typeNameC, typeNameD); | |
complexTypes.put(typeNameD, "uima.tcas.Annotation"); | |
List<TestFeature> listA = new ArrayList<RutaTestUtils.TestFeature>(); | |
features.put(typeNameA, listA); | |
String fnab = "ab"; | |
listA.add(new TestFeature(fnab, "", typeNameB)); | |
String fnac = "ac"; | |
listA.add(new TestFeature(fnac, "", typeNameC)); | |
List<TestFeature> listB = new ArrayList<RutaTestUtils.TestFeature>(); | |
features.put(typeNameB, listB); | |
String fnbc = "bc"; | |
listB.add(new TestFeature(fnbc, "", typeNameC)); | |
List<TestFeature> listC = new ArrayList<RutaTestUtils.TestFeature>(); | |
features.put(typeNameC, listC); | |
String fnci = "ci"; | |
listC.add(new TestFeature(fnci, "", "uima.cas.Integer")); | |
String fncb = "cb"; | |
listC.add(new TestFeature(fncb, "", "uima.cas.Boolean")); | |
List<TestFeature> listD = new ArrayList<RutaTestUtils.TestFeature>(); | |
features.put(typeNameD, listD); | |
String fnds = "ds"; | |
listD.add(new TestFeature(fnds, "", "uima.cas.String")); | |
CAS cas = RutaTestUtils.process(namespace + "/" + name + RutaEngine.SCRIPT_FILE_EXTENSION, | |
namespace + "/" + name + ".txt", 50, false, false, complexTypes, features, null); | |
RutaTestUtils.assertAnnotationsEquals(cas, 1, 1, "Marshall"); | |
RutaTestUtils.assertAnnotationsEquals(cas, 2, 1, "Marshall"); | |
RutaTestUtils.assertAnnotationsEquals(cas, 3, 1, "Marshall"); | |
RutaTestUtils.assertAnnotationsEquals(cas, 4, 1, "Marshall"); | |
RutaTestUtils.assertAnnotationsEquals(cas, 5, 3, "Peter", "Joern", "Marshall"); | |
RutaTestUtils.assertAnnotationsEquals(cas, 6, 3, "Kluegl", "Kottmann", "Schor"); | |
RutaTestUtils.assertAnnotationsEquals(cas, 7, 3, "Kluegl", "Kottmann", "Schor"); | |
RutaTestUtils.assertAnnotationsEquals(cas, 8, 3, "Kluegl", "Kottmann", "Schor"); | |
RutaTestUtils.assertAnnotationsEquals(cas, 9, 3, "Kluegl", "Kottmann", "Schor"); | |
RutaTestUtils.assertAnnotationsEquals(cas, 10, 1, "Marshall"); | |
RutaTestUtils.assertAnnotationsEquals(cas, 11, 1, "Marshall"); | |
RutaTestUtils.assertAnnotationsEquals(cas, 12, 1, "Marshall"); | |
RutaTestUtils.assertAnnotationsEquals(cas, 13, 3, "Peter", "Joern", "Marshall"); | |
RutaTestUtils.assertAnnotationsEquals(cas, 14, 3, "Kluegl", "Kottmann", "Schor"); | |
RutaTestUtils.assertAnnotationsEquals(cas, 15, 3, "Peter", "Joern", "Marshall"); | |
RutaTestUtils.assertAnnotationsEquals(cas, 16, 1, "Peter Kluegl, Joern Kottmann, Marshall"); | |
RutaTestUtils.assertAnnotationsEquals(cas, 17, 1, | |
"Peter Kluegl, Joern Kottmann, Marshall Schor"); | |
RutaTestUtils.assertAnnotationsEquals(cas, 18, 1, | |
"Peter Kluegl, Joern Kottmann, Marshall Schor"); | |
cas.release(); | |
} | |
} |