uimafit-legacy-support/src/test/java/org/apache/uima/fit/util/ContainmentIndexTest.java

/*
 * 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.fit.util;

import static java.util.Arrays.asList;
import static org.apache.uima.fit.util.JCasUtil.select;
import static org.apache.uima.fit.util.JCasUtil.selectCovered;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;

import java.util.ArrayList;
import java.util.List;

import org.apache.uima.fit.ComponentTestBase;
import org.apache.uima.fit.type.Sentence;
import org.apache.uima.fit.type.Token;
import org.apache.uima.fit.util.ContainmentIndex.Type;
import org.junit.Test;

/**
 * Unit test for {@link ContainmentIndex}.
 * 
 */
public class ContainmentIndexTest extends ComponentTestBase {
  @Test
  public void test() throws Exception {
    String text = "Will you come home today ? \n No , tomorrow !";
    tokenBuilder.buildTokens(jCas, text);

    List<Sentence> sentences = new ArrayList<Sentence>(select(jCas, Sentence.class));
    List<Token> tokens = new ArrayList<Token>(select(jCas, Token.class));

    ContainmentIndex<Sentence, Token> idx = ContainmentIndex.create(jCas, Sentence.class,
            Token.class, Type.BOTH);

    assertEquals(selectCovered(Token.class, sentences.get(0)), idx.containedIn(sentences.get(0)));
    assertEquals(selectCovered(Token.class, sentences.get(1)), idx.containedIn(sentences.get(1)));

    assertEquals(asList(sentences.get(0)), idx.containing(tokens.get(0)));
    assertEquals(asList(sentences.get(1)), idx.containing(tokens.get(tokens.size() - 1)));

    assertTrue(idx.isContainedIn(sentences.get(0), tokens.get(0)));
    assertFalse(idx.isContainedIn(sentences.get(0), tokens.get(tokens.size() - 1)));

    // After removing the annotation the index has to be rebuilt.
    assertTrue(idx.isContainedInAny(tokens.get(0)));
    sentences.get(0).removeFromIndexes();
    idx = ContainmentIndex.create(jCas, Sentence.class, Token.class, Type.BOTH);
    assertFalse(idx.isContainedInAny(tokens.get(0)));
  }
}